diff --git a/hugegraph-python/example/client_test.py b/hugegraph-python/example/client_test.py
index c3d980814..f68c727c9 100644
--- a/hugegraph-python/example/client_test.py
+++ b/hugegraph-python/example/client_test.py
@@ -2,7 +2,8 @@
 from typing import Any
 from unittest.mock import MagicMock, patch
 
-from example.test_hugegraph import HugeGraph
+from example.hugegraph_test import HugeGraph
+
 
 class HugeGraphTest(unittest.TestCase):
     def setUp(self) -> None: