Hotfix script

Made minor changes.