Fix the test I just broke