added dirty hack for wrong numbering in training data