diff --git a/crates/ui/src/components/stories/icon_button.rs b/crates/ui/src/components/stories/icon_button.rs
index ba3d5fd8660988298a38307cb8a690d1a365c7f4..4b53a2c0f230587fcce32edf5de734a41396f579 100644
--- a/crates/ui/src/components/stories/icon_button.rs
+++ b/crates/ui/src/components/stories/icon_button.rs
@@ -113,7 +113,7 @@ impl Render for IconButtonStory {
 
         StoryContainer::new(
             "Icon Button",
-            "crates/ui2/src/components/stories/icon_button.rs",
+            "crates/ui/src/components/stories/icon_button.rs",
         )
         .child(StorySection::new().children(buttons))
         .child(
diff --git a/crates/ui/src/components/stories/toggle_button.rs b/crates/ui/src/components/stories/toggle_button.rs
index da2a2512c44a7705d817a12a25b6355f7f792c04..68789a53409987c9582dd6b42027b342d1a93c97 100644
--- a/crates/ui/src/components/stories/toggle_button.rs
+++ b/crates/ui/src/components/stories/toggle_button.rs
@@ -9,7 +9,7 @@ impl Render for ToggleButtonStory {
     fn render(&mut self, _cx: &mut ViewContext<Self>) -> impl IntoElement {
         StoryContainer::new(
             "Toggle Button",
-            "crates/ui2/src/components/stories/toggle_button.rs",
+            "crates/ui/src/components/stories/toggle_button.rs",
         )
         .child(
             StorySection::new().child(