Attempt to fix the test