fixed the failing unit test