Fix feature selection not working