Adds missing import to util.py
Merge pull request #934 from remram44/util-missing-import

Adds missing import to util.py