diff --git a/src/array/cuda/dgl_cub.cuh b/src/array/cuda/dgl_cub.cuh
index 17dad7dc48f9..c68ea4d35d85 100644
--- a/src/array/cuda/dgl_cub.cuh
+++ b/src/array/cuda/dgl_cub.cuh
@@ -10,7 +10,9 @@
 // include cub in a safe manner
 #define CUB_NS_PREFIX namespace dgl {
 #define CUB_NS_POSTFIX }
+#define CUB_NS_QUALIFIER ::dgl::cub
 #include "cub/cub.cuh"
+#undef CUB_NS_QUALIFIER
 #undef CUB_NS_POSTFIX
 #undef CUB_NS_PREFIX