Changed a unit test slightly