Fixed the script