made the test a bit better