ooopps, swapped train and test