diff --git a/packages/react/src/theming.tsx b/packages/react/src/theming.tsx
index e6db452cf..dd513d033 100644
--- a/packages/react/src/theming.tsx
+++ b/packages/react/src/theming.tsx
@@ -58,7 +58,7 @@ const getTheme = (
 }
 
 let createCacheWithTheme = /* #__PURE__ */ weakMemoize((outerTheme: Theme) => {
-  return weakMemoize((theme: Theme) => {
+  return weakMemoize((theme: Partial<Theme> | ((theme: Theme) => Theme)) => {
     return getTheme(outerTheme, theme)
   })
 })