fixed the test a bit