diff --git a/Lib/test/dtracedata/call_stack.stp b/Lib/test/dtracedata/call_stack.stp
index d4455fc8489af2..2901f44654be37 100644
--- a/Lib/test/dtracedata/call_stack.stp
+++ b/Lib/test/dtracedata/call_stack.stp
@@ -10,15 +10,6 @@ function basename:string(path:string)
     return last_token;
 }
 
-probe @PYTHON_SYSTEMTAP_PROBE@("function__entry")
-{
-    funcname = user_string($arg2);
-
-    if (funcname == "start") {
-        tracing = 1;
-    }
-}
-
 probe @PYTHON_SYSTEMTAP_PROBE@("function__entry"),
       @PYTHON_SYSTEMTAP_PROBE@("function__return")
 {
@@ -26,17 +17,14 @@ probe @PYTHON_SYSTEMTAP_PROBE@("function__entry"),
     funcname = user_string($arg2);
     lineno = $arg3;
 
-    if (tracing) {
+    if (funcname == "start") {
+        if ($$name == "function__entry") {
+            tracing = 1;
+        } else {
+            tracing = 0;
+        }
+    } else if (tracing) {
         printf("%d\t%s:%s:%s:%d\n", gettimeofday_us(), $$name,
                basename(filename), funcname, lineno);
     }
 }
-
-probe @PYTHON_SYSTEMTAP_PROBE@("function__return")
-{
-    funcname = user_string($arg2);
-
-    if (funcname == "start") {
-        tracing = 0;
-    }
-}