Fix util.py