From 65d2c8cf831884b9bc76d069d10aa162bcefcc5a Mon Sep 17 00:00:00 2001
From: Peng Lyu <penn.lv@gmail.com>
Date: Mon, 13 May 2024 08:42:58 -0700
Subject: [PATCH] Fix #15666. Fix custom metadata.

---
 src/notebookRunGroups/util/cellMetadataHelpers.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/notebookRunGroups/util/cellMetadataHelpers.ts b/src/notebookRunGroups/util/cellMetadataHelpers.ts
index ee626a7..6bfbda6 100644
--- a/src/notebookRunGroups/util/cellMetadataHelpers.ts
+++ b/src/notebookRunGroups/util/cellMetadataHelpers.ts
@@ -4,7 +4,7 @@ import * as vscode from 'vscode';
 
 // Retrieve the cell run group value
 export function getCellRunGroupMetadata(cell: vscode.NotebookCell): string {
-    const customMetadata = cell.metadata?.custom?.metadata?.notebookRunGroups;
+    const customMetadata = useCustomMetadata() ? cell.metadata?.custom?.metadata?.notebookRunGroups : cell.metadata?.metadata?.notebookRunGroups;
 
     if (customMetadata && customMetadata.groupValue) {
         return customMetadata.groupValue;