diff --git a/src/app/components/generator/SchemaGenerator.tsx b/src/app/components/generator/SchemaGenerator.tsx
index dce031f3..43e45297 100644
--- a/src/app/components/generator/SchemaGenerator.tsx
+++ b/src/app/components/generator/SchemaGenerator.tsx
@@ -323,10 +323,8 @@ export function SchemaGenerator({ gen, allowedVersions }: Props) {
 		} else {
 			Analytics.showPreview(gen.id, 'menu')
 		}
+		Store.setPreviewPanelOpen(!previewShown)
 		setPreviewShown(!previewShown)
-		if (!previewShown && sourceShown) {
-			setSourceShown(false)
-		}
 	}
 
 	const [projectShown, setProjectShown] = useState(Store.getProjectPanelOpen() ?? false)