Fix unit test for util.py
Fix util.py to pass unit test