Fixing the script