diff --git a/src/core/impl/comp_node/cuda/comp_node.cpp b/src/core/impl/comp_node/cuda/comp_node.cpp
index ae296d98b729478cee86c6e92d38c7bba9fe0866..0d4b91c905804ab2259fbd33bd2d81810e56da2a 100644
--- a/src/core/impl/comp_node/cuda/comp_node.cpp
+++ b/src/core/impl/comp_node/cuda/comp_node.cpp
@@ -659,10 +659,10 @@ CompNode::Impl* CudaCompNode::load_cuda(
     if (!available_node) {
         mgb_assert(sd.nr_node < sd.MAX_NR_COMP_NODE,
                 "too many CompNode allocated");
-        mgb_assert(locator.device < sd.MAX_NR_COMP_NODE,
-                "device number too large");
         available_node = &sd.node[sd.nr_node ++];
     }
+    mgb_assert(locator.device < sd.MAX_NR_DEVICE,
+            "device number too large");
 
     mgb_assert(!available_node->m_initialized);
     available_node->init(locator, locator_logical);