- fix small return bug