diff --git a/packages/checkout/src/contexts/CheckoutModal.tsx b/packages/checkout/src/contexts/CheckoutModal.tsx
index a5a6a4bd..19849c9e 100644
--- a/packages/checkout/src/contexts/CheckoutModal.tsx
+++ b/packages/checkout/src/contexts/CheckoutModal.tsx
@@ -1,6 +1,6 @@
 import React from 'react'
 import { MethodArguments } from '../api'
-import { createGenericContext } from '@0xsequence/kit'
+import { createGenericContext, Theme } from '@0xsequence/kit'
 
 interface CoinQuantity {
   contractAddress: string,
@@ -41,7 +41,7 @@ type CheckoutModalContext = {
   triggerCheckout: (settings: CheckoutSettings) => void
   closeCheckout: () => void
   settings?: CheckoutSettings
-  theme: 'light'|'dark'
+  theme: Theme
 }
 
 export const [useCheckoutModalContext, CheckoutModalContextProvider] = createGenericContext<CheckoutModalContext>()
\ No newline at end of file
diff --git a/packages/kit/src/components/KitProvider/index.tsx b/packages/kit/src/components/KitProvider/index.tsx
index 30d7ace3..e18310c8 100644
--- a/packages/kit/src/components/KitProvider/index.tsx
+++ b/packages/kit/src/components/KitProvider/index.tsx
@@ -24,7 +24,7 @@ import '@0xsequence/design-system/styles.css'
 import * as sharedStyles from '../styles.css'
 
 export declare const THEME: readonly ["dark", "light"];
-export declare type Theme = ComponentProps<typeof ThemeProvider>['theme'];
+export declare type Theme = Exclude<ComponentProps<typeof ThemeProvider>['theme'], undefined>;
 export const THEMES = {
   dark: 'dark' as Theme,
   light: 'light' as Theme