Fixing old util package in main.py