make main.py work again