fix wrong output