fix return keyword bug
Merge pull request #1 from davidgristwood/master

fix return keyword bug