Utils.py -- minor bug fix

Needed to add some imports