diff --git a/crates/storybook2/src/collab_panel.rs b/crates/storybook2/src/collab_panel.rs
index 2f733c1b7600c0cf4fc510aca75fb0258034a8c1..a1efce7c0d235e29864552784a36ab4aaad994d8 100644
--- a/crates/storybook2/src/collab_panel.rs
+++ b/crates/storybook2/src/collab_panel.rs
@@ -1,8 +1,8 @@
-use crate::theme::{theme, Theme};
 use gpui3::{
     div, svg, view, AppContext, Context, Element, ElementId, IntoAnyElement, ParentElement,
     ScrollState, SharedString, StyleHelpers, Styled, View, ViewContext, WindowContext,
 };
+use ui::{theme, Theme};
 
 pub struct CollabPanel {
     scroll_state: ScrollState,
@@ -68,21 +68,21 @@ impl CollabPanel {
                             .py_2()
                             .flex()
                             .flex_col()
-                            .child(self.list_section_header(1, "CHANNELS", true, theme)),
+                            .child(self.list_section_header(1, "CHANNELS", true, &theme)),
                     )
                     .child(
                         div()
                             .py_2()
                             .flex()
                             .flex_col()
-                            .child(self.list_section_header(2, "CONTACTS", true, theme))
+                            .child(self.list_section_header(2, "CONTACTS", true, &theme))
                             .children(
                                 std::iter::repeat_with(|| {
                                     vec![
                                         self.list_item(
                                             "http://github.com/as-cii.png?s=50",
                                             "as-cii",
-                                            theme,
+                                            &theme,
                                         ),
                                         // self.list_item(
                                         //     "http://github.com/nathansobo.png?s=50",
diff --git a/crates/storybook2/src/stories/z_index.rs b/crates/storybook2/src/stories/z_index.rs
index 89a146b4ece86023bb47c5f33fab9d6bf3863e4f..4824386d8a32673196ebadf5b529699ef0d92c70 100644
--- a/crates/storybook2/src/stories/z_index.rs
+++ b/crates/storybook2/src/stories/z_index.rs
@@ -87,7 +87,7 @@ trait Styles: StyleHelpers {
     }
 }
 
-impl<S> Styles for Div<S> {}
+impl<S: 'static + Send + Sync> Styles for Div<S> {}
 
 #[derive(Element)]
 struct ZIndexExample<S: 'static + Send + Sync> {
diff --git a/crates/storybook2/src/storybook2.rs b/crates/storybook2/src/storybook2.rs
index 30fbb079a26699c70b9821cbf4313641c2dc8fc0..1f1a8af13d48aa798a772ac264a3de2eabe812d1 100644
--- a/crates/storybook2/src/storybook2.rs
+++ b/crates/storybook2/src/storybook2.rs
@@ -102,7 +102,7 @@ impl StoryWrapper {
         Self { selector, theme }
     }
 
-    fn render(&mut self, cx: &mut ViewContext<Self>) -> impl Element<State = Self> {
+    fn render(&mut self, cx: &mut ViewContext<Self>) -> impl Element<ViewState = Self> {
         themed(self.theme.clone(), cx, |cx| {
             div()
                 .flex()
diff --git a/crates/storybook2/src/workspace.rs b/crates/storybook2/src/workspace.rs
index e902df65653d57a9a305885801e3e4fd1bd5bc1b..2fb680cfc336907c81753901e2767659b3fce9c4 100644
--- a/crates/storybook2/src/workspace.rs
+++ b/crates/storybook2/src/workspace.rs
@@ -1,12 +1,12 @@
 use crate::{
     collab_panel::{collab_panel, CollabPanel},
-    theme::{theme, themed},
     themes::rose_pine,
 };
 use gpui3::{
     div, img, svg, view, Context, Element, ParentElement, RootView, StyleHelpers, Styled, View,
     ViewContext, WindowContext,
 };
+use ui::{theme, themed};
 
 pub struct Workspace {
     left_panel: View<CollabPanel, Self>,