Fix test that I missed