add missing returns. Fix #319
Merge pull request #335 from squirrelo/issue-319

Add missing returns. Fix #319