Fix issue 1
Merge pull request #2 from stac47/fix_issue_1

Fix issue 1