Fixed train model to do BagManipulation correctly