From 666438ea6118f792f56bae31eda62ce11b9917a9 Mon Sep 17 00:00:00 2001
From: Musale Martin <martinmusale@microsoft.com>
Date: Thu, 3 Aug 2023 14:38:53 +0300
Subject: [PATCH] Update pages that the theme toggle should not appear on

Signed-off-by: Musale Martin <martinmusale@microsoft.com>
---
 .storybook/addons/codeEditorAddon/codeAddon.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.storybook/addons/codeEditorAddon/codeAddon.js b/.storybook/addons/codeEditorAddon/codeAddon.js
index f9a11c894e..c3db8125cb 100644
--- a/.storybook/addons/codeEditorAddon/codeAddon.js
+++ b/.storybook/addons/codeEditorAddon/codeAddon.js
@@ -70,8 +70,9 @@ export const withCodeEditor = makeDecorator({
   skipIfNoParametersOrOptions: false,
   wrapper: (getStory, context, { options }) => {
     const forOptions = options ? options.disableThemeToggle : false;
-    const forContext =
-      context && (context.name === 'Custom CSS Properties' || context.title.toLowerCase().includes('templating'));
+    const title =
+      ['Custom CSS Properties', 'Theme'].includes(context.name) || context.title.toLowerCase().includes('templating');
+    const forContext = context && title;
     const disableThemeToggle = forOptions || forContext;
     let story = getStory(context);