small tweak in the test