diff --git a/crates/storybook2/src/stories/kitchen_sink.rs b/crates/storybook2/src/stories/kitchen_sink.rs
index 74e620fc31f0d108f0798edbd1e781fb6c3fd7eb..e0459098f574b84e5b4c6c3e7ab15a7860ead150 100644
--- a/crates/storybook2/src/stories/kitchen_sink.rs
+++ b/crates/storybook2/src/stories/kitchen_sink.rs
@@ -20,7 +20,7 @@ impl<S: 'static + Send + Sync + Clone> KitchenSinkStory<S> {
 
     fn render(&mut self, _view: &mut S, cx: &mut ViewContext<S>) -> impl Element<ViewState = S> {
         let element_stories = ElementStory::iter().map(|selector| selector.story());
-        let component_stories = ComponentStory::iter().map(|selector| selector.story(cx));
+        let component_stories = ComponentStory::iter().map(|selector| selector.story(cx)).collect::<Vec<_>>();
 
         Story::container(cx)
             .overflow_y_scroll(ScrollState::default())
diff --git a/crates/storybook2/src/story_selector.rs b/crates/storybook2/src/story_selector.rs
index bf169b72da280d03e1f0d2c4b8543cbca14f3bd2..19bd267a10482a650d65175f9880c32f9234e49a 100644
--- a/crates/storybook2/src/story_selector.rs
+++ b/crates/storybook2/src/story_selector.rs
@@ -90,7 +90,7 @@ impl ComponentStory {
             Self::Toast => ui::ToastStory::new().into_any(),
             Self::Toolbar => ui::ToolbarStory::new().into_any(),
             Self::TrafficLights => ui::TrafficLightsStory::new().into_any(),
-            Self::Workspace => ui::workspace_story(cx).into_any().into_any(),
+            Self::Workspace => todo!(),
         }
     }
 }
diff --git a/crates/storybook2/src/storybook2.rs b/crates/storybook2/src/storybook2.rs
index 1f1a8af13d48aa798a772ac264a3de2eabe812d1..87c94a8ef6d45f5d2d05bef6315f7abab0a0c6ed 100644
--- a/crates/storybook2/src/storybook2.rs
+++ b/crates/storybook2/src/storybook2.rs
@@ -108,7 +108,7 @@ impl StoryWrapper {
                 .flex()
                 .flex_col()
                 .size_full()
-                .child_any(self.selector.story())
+                .child_any(self.selector.story(cx))
         })
     }
 }