diff --git a/react/src/lib/components/SubsurfaceViewer/layers/wells/wellsLayer.stories.tsx b/react/src/lib/components/SubsurfaceViewer/layers/wells/wellsLayer.stories.tsx
index d0189c962..6920a06f0 100644
--- a/react/src/lib/components/SubsurfaceViewer/layers/wells/wellsLayer.stories.tsx
+++ b/react/src/lib/components/SubsurfaceViewer/layers/wells/wellsLayer.stories.tsx
@@ -571,7 +571,7 @@ const wellLayerTemplate = (args: any) => {
                     getInterpolateMethod={getInterpolateMethod}
                 />
             </div>
-            <DeckGLMap {...args} layers={layers} />
+            <SubsurfaceViewer {...args} layers={layers} />
         </div>
     );
 };
@@ -604,4 +604,4 @@ LegendWithColorSelector.parameters = {
         inlineStories: false,
         iframeHeight: 500,
     },
-};
+};
\ No newline at end of file