diff --git a/lib/custom-ui/components/layout.js b/lib/custom-ui/components/layout.js
index c22805997..10fb15b94 100644
--- a/lib/custom-ui/components/layout.js
+++ b/lib/custom-ui/components/layout.js
@@ -39,7 +39,7 @@ export const Layout = (props) => {
           if (test == false) {
             setStormLayers([...stormLayers, item.uniqueId])
             // setStormLayers2([...stormLayers2, item])
-            addStormLayers(viewState, item);
+            addStormLayers(props.viewState, item);
           }
         }
       });