diff --git a/heatmap-tool/frontend/src/components/MapDisplay.tsx b/heatmap-tool/frontend/src/components/MapDisplay.tsx
index b8e3ce65..382183cc 100644
--- a/heatmap-tool/frontend/src/components/MapDisplay.tsx
+++ b/heatmap-tool/frontend/src/components/MapDisplay.tsx
@@ -22,13 +22,9 @@ const DynamicLegendHandler = ({ onBoundsChange }: { onBoundsChange: (bounds: Lat
     const map = useMapEvents({
       zoomend: () => onBoundsChange(map.getBounds()),
       moveend: () => onBoundsChange(map.getBounds()),
+      load: () => onBoundsChange(map.getBounds()), // Handle initial load
     });
   
-    // Initial load
-    useEffect(() => {
-        onBoundsChange(map.getBounds());
-    }, []); // Run only once on initial mount
-  
     return null;
 };