Fix for the script