Fixed missed import in the utils.py