Fix the test case I just committed