diff --git a/packages/vscode/src/docsnotebook.ts b/packages/vscode/src/docsnotebook.ts
index 910279c1b..34975e3e1 100644
--- a/packages/vscode/src/docsnotebook.ts
+++ b/packages/vscode/src/docsnotebook.ts
@@ -104,12 +104,9 @@ function activateNotebookExecutor(state: ExtensionState) {
         await state.parseWorkspace()
         const project = state.project
 
-        // frontmatter
         const firstCell = notebook.cellAt(0)
         const { genaiscript: frontmatter = {} } =
-            (firstCell.document.languageId === "yaml" &&
-                YAMLTryParse(firstCell.document.getText())) ||
-            {}
+            frontmatterTryParse(firstCell?.document?.getText())?.value || {}
         const {
             model,
             files,