diff --git a/ui-spacetimechart/src/stories/rectangle-zoom.stories.tsx b/ui-spacetimechart/src/stories/rectangle-zoom.stories.tsx
index a7bc7284..572d59d2 100644
--- a/ui-spacetimechart/src/stories/rectangle-zoom.stories.tsx
+++ b/ui-spacetimechart/src/stories/rectangle-zoom.stories.tsx
@@ -8,8 +8,8 @@ import '@osrd-project/ui-core/dist/theme.css';
 import '../styles/stories/rectangle-zoom.css';
 
 import { OPERATIONAL_POINTS, PATHS } from './lib/paths';
+import { CanvasRect } from '../components/CanvasRect';
 import { PathLayer } from '../components/PathLayer';
-import { SimpleRect } from '../components/SimpleRect';
 import { SpaceTimeChart } from '../components/SpaceTimeChart';
 import { CAPTION_SIZE } from '../components/TimeCaptions';
 import { type Point, type PathData, type OperationalPoint } from '../lib/types';
@@ -313,7 +313,7 @@ const RectangleZoomWrapper = ({
           <PathLayer key={path.id} path={path} color={path.color} />
         ))}
         {state.rect && (
-          <SimpleRect
+          <CanvasRect
             timeStart={state.rect.timeStart}
             timeEnd={state.rect.timeEnd}
             spaceStart={state.rect.spaceStart}