diff --git a/app/assets/javascripts/editor/execution.js.erb b/app/assets/javascripts/editor/execution.js.erb
index 53b53ff7..c28c47a4 100644
--- a/app/assets/javascripts/editor/execution.js.erb
+++ b/app/assets/javascripts/editor/execution.js.erb
@@ -17,11 +17,13 @@ CodeOceanEditorWebsocket = {
   initializeSocketForTesting: function(url) {
     this.initializeSocket(url);
     this.websocket.on('default',this.handleTestResponse.bind(this));
+    this.websocket.on('exit', this.handleExitCommand.bind(this));
   },
 
   initializeSocketForScoring: function(url) {
     this.initializeSocket(url);
-    this.websocket.on('default',this.handleScoringResponse.bind(this))
+    this.websocket.on('default',this.handleScoringResponse.bind(this));
+    this.websocket.on('exit', this.handleExitCommand.bind(this));
   },
 
   initializeSocketForRunning: function(url) {