Fix training data