Fixes in util.py