Fix wrong output