removed a bug with the script