diff --git a/systrace/src/trace/systrace_manager.cc b/systrace/src/trace/systrace_manager.cc
index 783d25445fc691173f2fbb486ae679218bb2abc8..0996ee92a3db145d5d7858d764b6f21dbeec2969 100644
--- a/systrace/src/trace/systrace_manager.cc
+++ b/systrace/src/trace/systrace_manager.cc
@@ -99,7 +99,7 @@ void PyTorchTrace::dumpPyTorchTracing()
 
     std::lock_guard<std::mutex> lock(trace_mutex_);
 
-    pytorch_trace_.set_rank(config::GlobalConfig::Instance().local_rank);
+    pytorch_trace_.set_rank(config::GlobalConfig::Instance().rank);
     pytorch_trace_.set_comm(config::GlobalConfig::Instance().job_name);
 
     for (size_t i = 0; i < pytorch_tracing_functions_.size(); ++i)
@@ -272,4 +272,4 @@ void SysTrace::eventPollerMain()
     }
 }
 
-} // namespace systrace
\ No newline at end of file
+} // namespace systrace