diff --git a/demo/src/main.tsx b/demo/src/main.tsx
index 70e7abb..a722749 100644
--- a/demo/src/main.tsx
+++ b/demo/src/main.tsx
@@ -14,10 +14,8 @@ import { ThemeContext } from './common/ThemeContext';
 
 export const CustomThemeProvider = ({
   children,
-  className,
 }: {
   children: React.ReactNode;
-  className?: string;
 }) => {
   const [theme, setTheme] = React.useState<ThemeType>(() =>
     window.matchMedia('(prefers-color-scheme: dark)').matches
@@ -27,7 +25,7 @@ export const CustomThemeProvider = ({
 
   return (
     <ThemeContext.Provider value={{ theme, setTheme }}>
-      <ThemeProvider theme={theme} className={className}>
+      <ThemeProvider theme={theme} className='custom-theme'>
         {children}
       </ThemeProvider>
     </ThemeContext.Provider>
@@ -37,7 +35,7 @@ export const CustomThemeProvider = ({
 createRoot(document.getElementById('root') as HTMLElement).render(
   <React.StrictMode>
     <HashRouter>
-      <CustomThemeProvider className='custom-theme'>
+      <CustomThemeProvider>
         <Routes>
           <Route path='/' element={<App />} />
           {DEMOS_LIST.map(({ layouts }) =>