fix KO return bug