Forgot to update the test script