Fixing the test again