[BUGFIX] Enable localizationStateSelector for selectTreeElement