Auto setting of category on setting primary

Before FlexiContent Stable, when you create a new content item and set the primary category, the same category is automatically selected in the list of categories. In the new Stable version, you have to do this manually. Would it be possible again to have the primary category selected in the list of categories automatically when it is selected in the primary category select box?