Fix error in the script