fixed the failing test case