Fix for LearningAgentTest test_getTrainingData()