Fixed return bug.