diff --git a/src/event_handlers/source_code_event_handler.py b/src/event_handlers/source_code_event_handler.py
index 2b86766..942a9e1 100644
--- a/src/event_handlers/source_code_event_handler.py
+++ b/src/event_handlers/source_code_event_handler.py
@@ -75,6 +75,7 @@ class SourceCodeEventHandler(EventHandlerBase, SupervisorMixin):
 
     def write(self, data_json):
         self.filemanager.file_contents = data_json['data']['content']
+        data_json['data']['files'] = self.filemanager.files
         self.restart_process()
 
     def select(self, data_json):