diff --git a/tests/pytest/util/common.py b/tests/pytest/util/common.py
index 3d25207158..63dc9773c4 100644
--- a/tests/pytest/util/common.py
+++ b/tests/pytest/util/common.py
@@ -29,6 +29,7 @@ from util.constant import *
 from dataclasses import dataclass,field
 from typing import List
 from datetime import datetime, timedelta
+import re
 
 @dataclass
 class DataSet: