Fix exception tests in train-test eval test