fixes failing test cases