diff --git a/styles/src/component/tab_bar_button.ts b/styles/src/component/tab_bar_button.ts
index 2b657e8b37ff2a4860a51e05dc003c2d3f77368c..0c43e7010e5469c10f959e00f4df8d177963392f 100644
--- a/styles/src/component/tab_bar_button.ts
+++ b/styles/src/component/tab_bar_button.ts
@@ -1,4 +1,4 @@
-import { ColorScheme, StyleSets } from "../common"
+import { Theme, StyleSets } from "../common"
 import { interactive } from "../element"
 import { InteractiveState } from "../element/interactive"
 import { background, foreground } from "../style_tree/components"
@@ -12,7 +12,7 @@ type TabBarButtonProps = TabBarButtonOptions & {
     state?: Partial<Record<InteractiveState, Partial<TabBarButtonOptions>>>
 }
 
-export function tab_bar_button(theme: ColorScheme, { icon, color = "base" }: TabBarButtonProps) {
+export function tab_bar_button(theme: Theme, { icon, color = "base" }: TabBarButtonProps) {
     const button_spacing = 8
 
     return (