Fix problem in test_util