Update util.py

Fix for #15