code review
fix failing test case