remove some expects from train_spec