diff --git a/services/web/app/views/project/editor.pug b/services/web/app/views/project/editor.pug
index 36b87d1bc7..44d19e2bea 100644
--- a/services/web/app/views/project/editor.pug
+++ b/services/web/app/views/project/editor.pug
@@ -92,7 +92,7 @@ block content
 					ng-class="{ 'ide-history-open' : (ui.view == 'history' && history.isV2) }",
 					layout="main",
 					ng-hide="state.loading",
-					resize-on="layout:chat:resize,history:toggle",
+					resize-on="layout:chat:resize,history:toggle,layout:flat-screen:toggle",
 					minimum-restore-size-west="130"
 					custom-toggler-pane=hasFeature('custom-togglers') ? "west" : false
 					custom-toggler-msg-when-open=hasFeature('custom-togglers') ? translate("tooltip_hide_filetree") : false
diff --git a/services/web/frontend/js/ide/directives/layout.js b/services/web/frontend/js/ide/directives/layout.js
index d340e9f5f9..448a3cc46f 100644
--- a/services/web/frontend/js/ide/directives/layout.js
+++ b/services/web/frontend/js/ide/directives/layout.js
@@ -162,7 +162,9 @@ export default App.directive('layout', ($parse, $compile, ide) => ({
             }
           }
 
-          return element.layout().resizeAll()
+          ide.$timeout(() => {
+            element.layout().resizeAll()
+          })
         }
 
         element.layout(options)