Forgot to update test.