Fixes main_script.py