Fix minor bug in SimpleTrainingImageSource