Fix test, unfortunately haven't tracked when I broke it