diff --git a/tests/python_client/testcases/test_partition.py b/tests/python_client/testcases/test_partition.py
index 5b3b92755b..e9a5a007cd 100644
--- a/tests/python_client/testcases/test_partition.py
+++ b/tests/python_client/testcases/test_partition.py
@@ -3,7 +3,7 @@ import logging
 from multiprocessing import Pool, Process
 import pytest
 from utils.utils import *
-from common.constants import *
+from common.constants import default_entities, default_fields
 from common.common_type import CaseLabel
 
 TIMEOUT = 120