Update util.py

Removed unused import