fixed error in main_test.py caused by main.py