diff --git a/extensions/notebook/src/book/bookTreeItem.ts b/extensions/notebook/src/book/bookTreeItem.ts
index d24809619a..25ecc49d0d 100644
--- a/extensions/notebook/src/book/bookTreeItem.ts
+++ b/extensions/notebook/src/book/bookTreeItem.ts
@@ -45,7 +45,7 @@ export class BookTreeItem extends vscode.TreeItem {
 				this.contextValue = 'savedBook';
 			}
 		} else {
-			if (book.type === BookTreeItemType.Markdown && book.page.expand_sections) {
+			if (book.page && book.page.sections && book.page.sections.length > 0) {
 				this.contextValue = 'section';
 			}
 			this.setPageVariables();