Forgot to fix these tests