fix the return bug.