changed the test a little bit