diff --git a/tests/python_client/testcases/test_flush.py b/tests/python_client/testcases/test_flush.py
index 6fb82fb188..b38096b1f4 100644
--- a/tests/python_client/testcases/test_flush.py
+++ b/tests/python_client/testcases/test_flush.py
@@ -1,6 +1,6 @@
 import pytest
 from utils.utils import *
-from common.constants import *
+from common.constants import default_entities
 from common.common_type import CaseLabel
 
 DELETE_TIMEOUT = 60