diff --git a/pyinfinitensor/src/pyinfinitensor/__init__.py b/pyinfinitensor/src/pyinfinitensor/__init__.py
index c9558bfe..ee95cb9d 100644
--- a/pyinfinitensor/src/pyinfinitensor/__init__.py
+++ b/pyinfinitensor/src/pyinfinitensor/__init__.py
@@ -1,7 +1,6 @@
 import sys
-from pathlib import Path
 
-sys.path.append(str(Path(str(__file__)).parent))
+sys.path.extend(__path__)
 
 import backend