Fixes issue with return nil
Merge pull request #10 from SteveyPugs/master

Fixes issue with return nil