fix 3 out of 4 test cases