diff --git a/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp b/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp
index 83a850aa3..40c452110 100644
--- a/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp
+++ b/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp
@@ -230,8 +230,7 @@ class NeighborSampler {
           add(row_start + index_data[i], global_src_node, local_src_node,
               dst_mapper, out_global_dst_nodes);
         }
-      }
-      else {
+      } else {
         const auto rand = at::empty_like(weight).uniform_();
         const auto a = (rand.log() / weight);
         const auto index = std::get<1>(a.topk(count));