diff --git a/kernels/lib_tree_filter/setup.py b/kernels/lib_tree_filter/setup.py
index a4b6398..6ad74ac 100644
--- a/kernels/lib_tree_filter/setup.py
+++ b/kernels/lib_tree_filter/setup.py
@@ -31,7 +31,7 @@
             name='tree_filter_cuda',
             include_dirs=[extensions_dir],
             sources=sources,
-            library_dirs=cuda_lib_path,
+            library_dirs=include_paths(cuda=True),
             extra_compile_args={'cxx':['-O3'],
                                 'nvcc':['-O3']})
     ],