fix the wrong output