issue #23

Fixed for training files too