fixes for feature selection