diff --git a/src/cpp/src/data/samplers/neighbor.cpp b/src/cpp/src/data/samplers/neighbor.cpp
index 8403c3b7..7ed6bf5b 100644
--- a/src/cpp/src/data/samplers/neighbor.cpp
+++ b/src/cpp/src/data/samplers/neighbor.cpp
@@ -496,7 +496,7 @@ DENSEGraph LayeredNeighborSampler::getNeighbors(torch::Tensor node_ids, shared_p
         }
 
         if (outgoing_offsets.defined()) {
-            if (delta_outgoing_edges.size(0) > 0) {
+            if (delta_outgoing_offsets.size(0) > 0) {
                 outgoing_offsets = outgoing_offsets + delta_outgoing_edges.size(0);
                 outgoing_offsets = torch::cat({delta_outgoing_offsets, outgoing_offsets}, 0);
             }