diff --git a/python/triton_tvm_ffi/utils.py b/python/triton_tvm_ffi/utils.py
index 694b438..87d7e34 100644
--- a/python/triton_tvm_ffi/utils.py
+++ b/python/triton_tvm_ffi/utils.py
@@ -1,13 +1,12 @@
-import importlib.resources
-from importlib.resources.abc import Traversable
+import sysconfig
 from typing import List, Optional
 
 from triton.backends.nvidia.driver import ty_to_cpp
 
 
 def include_paths() -> List[str]:
-    pkg_path: Traversable = importlib.resources.files("triton_tvm_ffi")
-    return [str(pkg_path / "include"), str(pkg_path / "../../include")]
+    pkg_path: str = sysconfig.get_path("purelib")
+    return [f"{pkg_path}/triton_tvm_ffi/include"]
 
 
 def type_canonicalize(ty: str) -> Optional[str]: