diff --git a/src/plugins/GenerateJob/templates/main.ejs b/src/plugins/GenerateJob/templates/main.ejs
index 19bedb605..1ed28d284 100644
--- a/src/plugins/GenerateJob/templates/main.ejs
+++ b/src/plugins/GenerateJob/templates/main.ejs
@@ -1,5 +1,15 @@
 import json
 import deepforge
+import signal
+import sys
+
+def signal_handler(signum, frame):
+    sig = signal.Signals(signum)
+    print('Received ' + sig.name + '. Exiting.')
+    sys.exit(1)
+
+signal.signal(signal.SIGINT, signal_handler)
+signal.signal(signal.SIGTERM, signal_handler)
 
 <%= initCode %>
 # input data<% inputs.forEach(function(pair) { var input = pair[0], type = pair[1], isNil = pair[2];%>