initial fix commit
Merge pull request #4 from BennyJake/issue-#3-fix

Fix for Issue #3