fixed test 1 failed: case 2: expected 6 got 4