ExtractTrainingFeatures bugfix