Fix the test i just broke