Added missing import from utils.py