Fix test that I broke