diff --git a/frontend/src/VectorVisualizer.tsx b/frontend/src/VectorVisualizer.tsx
index df31080..0cc4661 100644
--- a/frontend/src/VectorVisualizer.tsx
+++ b/frontend/src/VectorVisualizer.tsx
@@ -1,4 +1,4 @@
-import React, { useEffect, useState } from 'react';
+import React, { useEffect, useState, useRef } from 'react';
 import Box from '@mui/material/Box';
 import Card from '@mui/material/Card';
 import Typography from '@mui/material/Typography';
@@ -107,6 +107,7 @@ const VectorVisualizer: React.FC<VectorVisualizerProps> = (props: VectorVisualiz
   const [newQueryEmbedding, setNewQueryEmbedding] = useState<ChromaResult | undefined>(undefined);
   const [result, setResult] = useState<ResultData | undefined>(undefined);
   const [view2D, setView2D] = useState<boolean>(true);
+  const plotlyRef = useRef(null);
   const [tooltip, setTooltip] = useState<{
     visible: boolean,
     // x: number,
@@ -145,6 +146,9 @@ const VectorVisualizer: React.FC<VectorVisualizerProps> = (props: VectorVisualiz
   /* Trigger a resize event to force Plotly to rescale */
   useEffect(() => {
     window.dispatchEvent(new Event('resize'));
+    if (plotlyRef.current) {
+      Plot.Plots.resize(plotlyRef.current);
+    }
   }, [])
 
   useEffect(() => {
@@ -328,15 +332,17 @@ const VectorVisualizer: React.FC<VectorVisualizerProps> = (props: VectorVisualiz
       <FormControlLabel
         sx={{
           display: "flex",
-          position: "absolute",
+          position: "relative",
           width: "fit-content",
           ml: 1,
+          mb: '-2.5rem',
           zIndex: 100,
           flexBasis: 0,
           flexGrow: 0
         }}
         control={<Switch checked={!view2D} />} onChange={() => setView2D(!view2D)} label="3D" />
       <Plot
+        ref={plotlyRef}
         onClick={(event: any) => {
           const point = event.points[0];
           console.log('Point:', point);