Fix script
Merge pull request #61 from bhatfield/master

Fix script