Fixing the test for the better