diff --git a/crates/theme2/src/styles/stories/color.rs b/crates/theme2/src/styles/stories/color.rs
index a7d8885848ba89ed1470b6624a00dd56c39ade51..f406f64cec839e79f6ac4a6519165e428bba1930 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 237f2f10817872ea07c0f7fafa5745ee8722cf13..d7ab054b2d32f792f95f5ba9001209ca326d03ed 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()