Rework unused import in test_utils.py.