Fix the test that was failing