Fix script.
Merge pull request #437 from gsn/master

Fix script.