Fix error in utils.py