fixing the failing test case