Had to fix the script