diff --git a/MANIFEST.in b/MANIFEST.in
index 927383884..450750eae 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -12,6 +12,11 @@ recursive-exclude third_party/parallel-hashmap/tests *
 recursive-exclude third_party/parallel-hashmap/examples *
 recursive-exclude third_party/parallel-hashmap/benchmark *
 recursive-exclude third_party/METIS/graphs *
+recursive-exclude third_party/cccl/benchmarks *
+recursive-exclude third_party/cccl/docs *
+recursive-exclude third_party/cccl/examples *
+recursive-exclude third_party/cuCollections/examples *
+recursive-exclude third_party/cuCollections/tests *
 
 recursive-exclude test *
 recursive-exclude docs *
diff --git a/pyg_lib/csrc/ops/cuda/sampled_kernel.cu b/pyg_lib/csrc/ops/cuda/sampled_kernel.cu
index ae916dc3f..e36a83a32 100644
--- a/pyg_lib/csrc/ops/cuda/sampled_kernel.cu
+++ b/pyg_lib/csrc/ops/cuda/sampled_kernel.cu
@@ -2,15 +2,13 @@
 #include <ATen/cuda/CUDAContext.h>
 #include <torch/library.h>
 
-#include <cuco/static_map.cuh>
-
 namespace pyg {
 namespace ops {
 
 namespace {
 
 #define THREADS 1024
-#define CDIV(N, M) ((N) + (M) - 1) / (M)
+#define CDIV(N, M) ((N) + (M)-1) / (M)
 
 enum FnType { ADD, SUB, MUL, DIV };
 const std::map<std::string, FnType> to_fn_type = {