diff --git a/script/local.py b/script/local.py
index 41d723c..926b498 100644
--- a/script/local.py
+++ b/script/local.py
@@ -160,4 +160,8 @@ def get_pipeline_msg():
         data = stream.receive()
         if not data: 
             raise Exception("Received None message.")
-        return data.data    
\ No newline at end of file
+        return data.data    
+
+
+def on_system_restart(path):   
+    stop_pipeline()     
\ No newline at end of file