1 files changed, 14 insertions, 3 deletions

diff --git a/subprojects/frontend/src/editor/GenerateButton.tsx b/subprojects/frontend/src/editor/GenerateButton.tsx
index 4b998af6..a28f6b4b 100644
--- a/subprojects/frontend/src/editor/GenerateButton.tsx
+++ b/subprojects/frontend/src/editor/GenerateButton.tsx
@@ -3,12 +3,23 @@ import Button from '@mui/material/Button';
 import { observer } from 'mobx-react-lite';
 import React from 'react';
 
-import { useRootStore } from '../RootStore';
+import type EditorStore from './EditorStore';
 
 const GENERATE_LABEL = 'Generate';
 
-function GenerateButton(): JSX.Element {
-  const { editorStore } = useRootStore();
+function GenerateButton({
+  editorStore,
+}: {
+  editorStore: EditorStore | undefined;
+}): JSX.Element {
+  if (editorStore === undefined) {
+    return (
+      <Button color="inherit" className="rounded" disabled>
+        Loading&hellip;
+      </Button>
+    );
+  }
+
   const { errorCount, warningCount } = editorStore;
 
   const diagnostics: string[] = [];