Fixed the script to run locally