diff --git a/docs/snippets/common/button-story-project-args-theme.js.mdx b/docs/snippets/common/button-story-project-args-theme.js.mdx
index da897c5a7829..c3fd8bd5e666 100644
--- a/docs/snippets/common/button-story-project-args-theme.js.mdx
+++ b/docs/snippets/common/button-story-project-args-theme.js.mdx
@@ -2,7 +2,7 @@
 // preview.js
 
 // All stories expect a theme arg
-export const argTypes = { theme: { control: { options: ['light', 'dark'] } } };
+export const argTypes = { theme: { control: 'select', options: ['light', 'dark'] } };
 
 // The default value of the theme arg to all stories
 export const args = { theme: 'light' };