Forgot to fix this test as well.