Forgot this part of the test fix