fixing a little bug that changes the count of features from testing to training