And change the test case to match