Fixed one of the training data equations