Fix the test I just added