Fix tiny issue with overincluding files in the training data