fixed non working return
Merge pull request #1 from let4be/master

fixed non working return