Forgot to fix a couple of my test changes