update script to fix issue #23
Merge pull request #24 from BigManApproved/issue-23

update script to fix issue #23