fix the failing unit test