Tweak the test for now