fix traindata