Minor script fix
Merge pull request #2 from miludmann/master

Minor script fix