Update utils.py

Minor bug fix