Forgot one test to fix