Fixing the script
I don't really know