Fixed a minor issue in script