Fix wrong imports in utils.py