From 22560c475e9b78507d9fb353b35c5c599e33c99f Mon Sep 17 00:00:00 2001
From: Michael Shanks <mike@budibase.com>
Date: Tue, 25 Feb 2020 10:01:26 +0000
Subject: [PATCH] bugfix: create new screen exception

---
 packages/builder/src/builderStore/store.js | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/packages/builder/src/builderStore/store.js b/packages/builder/src/builderStore/store.js
index 1bc2019190..4d34b7e355 100644
--- a/packages/builder/src/builderStore/store.js
+++ b/packages/builder/src/builderStore/store.js
@@ -473,9 +473,12 @@ const _saveScreen = async (store, s, screen) => {
         ),
         updatedScreen,
       ]
-      s.pages[s.currentPageName]._screens = screens
-      s.screens = screens
-      _savePage(s)
+      store.update(innerState => {
+        innerState.pages[s.currentPageName]._screens = screens
+        innerState.screens = screens
+        _savePage(innerState)
+        return innerState
+      })
     })
 
   return s
@@ -502,7 +505,9 @@ const createScreen = store => (screenName, route, layoutComponentName) => {
     s.currentComponentInfo = newScreen.props
     s.currentFrontEndType = "screen"
 
-    return _saveScreen(store, s, newScreen)
+    _saveScreen(store, s, newScreen)
+
+    return s
   })
 }