Fix useTheme hook type

Temzasse · Mar 17, 2022 · 5fb9943 · 5fb9943
1 parent 8b3f0d3
commit 5fb9943
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/types/stitches.d.ts b/src/types/stitches.d.ts
@@ -48,7 +48,11 @@ export default interface Stitches<
   useTheme: () => string &
     {
       [Scale in keyof Theme]: {
-        [Token in keyof Theme[Scale]]: number | string;
+        [Token in keyof Theme[Scale]]: ThemeUtil.Token<
+          Extract<Token, string | number>,
+          string,
+          Extract<Scale, string | void>
+        >;
       };
     };
   ThemeProvider: React.FunctionComponent<{ theme?: any }>; // TODO: fix `any`