fix the code, not the test