Fixing broken import in utils.py