Update util_test.py