diff --git a/src/components/TrainingPanel.tsx b/src/components/TrainingPanel.tsx
index 11bca65..fb7220b 100644
--- a/src/components/TrainingPanel.tsx
+++ b/src/components/TrainingPanel.tsx
@@ -125,8 +125,7 @@ export default function TrainingPanel() {
         setActiveRunId(null);
       }
     }).finally(() => setIsLoadingRuns(false));
-  // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [expanded]);
+  }, [expanded, fetchDatasetInfo, fetchRuns, fetchActiveRun]);
 
   // Poll while training is active
   useEffect(() => {