diff --git a/extension_cpp_stable/extension_cpp_stable/__init__.py b/extension_cpp_stable/extension_cpp_stable/__init__.py
index fb8aad8..6bf1f58 100644
--- a/extension_cpp_stable/extension_cpp_stable/__init__.py
+++ b/extension_cpp_stable/extension_cpp_stable/__init__.py
@@ -1 +1,3 @@
+import torch
+from pathlib import Path
 from . import _C, ops  # noqa: F401