Fixing the test