Update _util.py

Fixing missing import.