fixed test 1 case 2: expected 4 got 6