diff --git a/debug/accuracy_tools/msprobe/pytorch/online_dispatch/utils.py b/debug/accuracy_tools/msprobe/pytorch/online_dispatch/utils.py
index 52b6a637b902a2358a5897fe018099f457a0bbcb..cf3605180991cec5067e754098a688eef71a48bd 100644
--- a/debug/accuracy_tools/msprobe/pytorch/online_dispatch/utils.py
+++ b/debug/accuracy_tools/msprobe/pytorch/online_dispatch/utils.py
@@ -59,10 +59,7 @@ INT_TYPE = [np.int32, np.int64]
 def get_callstack():
     callstack = []
     for (_, path, line, func, code, _) in inspect.stack()[2:]:
-        if code:
-            stack_line = [path, str(line), func, code[0].strip() if code else code]
-        else:
-            stack_line = [path, str(line), func, code]
+        stack_line = [path, str(line), func, code[0].strip() if code else code]
         callstack.append(stack_line)
     return callstack