fix a return bug

fix a return bug