Fix the failing unit test.