Updating script per issue #3

Fixing boo-boo