diff --git a/dsv4/kernels/cache/gather.py b/dsv4/kernels/cache/gather.py
index e2122bee..5b497dd3 100644
--- a/dsv4/kernels/cache/gather.py
+++ b/dsv4/kernels/cache/gather.py
@@ -15,7 +15,7 @@ import torch
 from torch.utils.cpp_extension import load
 
 _kernel_module = None
-_kernel_dir = os.path.join(os.path.dirname(__file__), "cuda")
+_kernel_dir = os.path.join(os.path.dirname(__file__), "..", "cuda")
 
 
 def _get_kernel_module():