updated tests to throw train exception