diff --git a/src/app/ide/ide.component.ts b/src/app/ide/ide.component.ts
index 7b8ad82..1a6b8c4 100644
--- a/src/app/ide/ide.component.ts
+++ b/src/app/ide/ide.component.ts
@@ -94,6 +94,7 @@ export class IdeComponent implements OnInit {
       }
       this.code = (message.content != null) ? message.content : this.code;
       this.files = message.files;
+      this.autoDetectEditorLanguage(this.filename);
     }
   }
 
@@ -141,7 +142,6 @@ export class IdeComponent implements OnInit {
     }
     this.filename = file;
     this.requestCode();
-    this.autoDetectEditorLanguage(this.filename);
   }
 
   editorWriteHandler() {