diff --git a/src/torch_ucc_tracing.cpp b/src/torch_ucc_tracing.cpp
index 994592e..1106b42 100644
--- a/src/torch_ucc_tracing.cpp
+++ b/src/torch_ucc_tracing.cpp
@@ -177,6 +177,8 @@ void CommTraceLogger::recordComms(
       dtype,
       curInSplitSizes_,
       curOutSplitSizes_,
+      0,
+      0,
       world_size);
 
   ++seqnum;