add missing return, thanks to iyuan for noticing