diff --git a/crates/theme2/src/styles/stories/color.rs b/crates/theme2/src/styles/stories/color.rs
index a7d8885848ba8..f406f64cec839 100644
--- a/crates/theme2/src/styles/stories/color.rs
+++ b/crates/theme2/src/styles/stories/color.rs
@@ -1,5 +1,5 @@
 use gpui::prelude::*;
-use gpui::{div, px, Div, ViewContext};
+use gpui::{div, px, ViewContext};
 use story::Story;
 
 use crate::{default_color_scales, ColorScaleStep};
@@ -7,9 +7,7 @@ use crate::{default_color_scales, ColorScaleStep};
 pub struct ColorsStory;
 
 impl Render for ColorsStory {
-    type Element = Div;
-
-    fn render(&mut self, cx: &mut ViewContext<Self>) -> Self::Element {
+    fn render(&mut self, cx: &mut ViewContext<Self>) -> impl Element {
         let color_scales = default_color_scales();
 
         Story::container().child(Story::title("Colors")).child(
diff --git a/crates/theme2/src/styles/stories/players.rs b/crates/theme2/src/styles/stories/players.rs
index 237f2f1081787..d7ab054b2d32f 100644
--- a/crates/theme2/src/styles/stories/players.rs
+++ b/crates/theme2/src/styles/stories/players.rs
@@ -1,4 +1,4 @@
-use gpui::{div, img, px, Div, ParentElement, Render, Styled, ViewContext};
+use gpui::{div, img, px, Element, ParentElement, Render, Styled, ViewContext};
 use story::Story;
 
 use crate::{ActiveTheme, PlayerColors};
@@ -6,9 +6,7 @@ use crate::{ActiveTheme, PlayerColors};
 pub struct PlayerStory;
 
 impl Render for PlayerStory {
-    type Element = Div;
-
-    fn render(&mut self, cx: &mut ViewContext<Self>) -> Self::Element {
+    fn render(&mut self, cx: &mut ViewContext<Self>) -> impl Element {
         Story::container().child(
             div()
                 .flex()