1 files changed, 3 insertions, 0 deletions
diff --git a/runtime/io.c b/runtime/io.c
index b72590f3..5c7a5f10 100644
--- a/runtime/io.c
+++ b/runtime/io.c
@@ -116,6 +116,9 @@ void _stp_warn (const char *fmt, ...)
  */
 void _stp_exit (void)
 {
+#ifdef SYSTEMTAP
+	atomic_set (&session_state, STAP_SESSION_STOPPING);
+#endif
 	_stp_exit_flag = 1;
 }