need to fix the script