Update public/components/MDSEnabledComponent/MDSEnabledComponent.tsx

Co-authored-by: SuZhou-Joe <[email protected]> Signed-off-by: Ramakrishna Chilaka <[email protected]> Signed-off-by: Ramakrishna Chilaka <[email protected]>
RamakrishnaChilaka · Mar 26, 2024 · 2c51c7a · 2c51c7a
1 parent f9018de
commit 2c51c7a
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/public/components/MDSEnabledComponent/MDSEnabledComponent.tsx b/public/components/MDSEnabledComponent/MDSEnabledComponent.tsx
@@ -37,16 +37,15 @@ export default class MDSEnabledComponent<
 export function useUpdateUrlWithDataSourceProperties() {
   const dataSourceMenuProps = useContext(DataSourceMenuContext);
   const { dataSourceId, dataSourceLabel, multiDataSourceEnabled } = dataSourceMenuProps;
-  if (multiDataSourceEnabled) {
-    // mds flag can't change while the app is loaded
-    const history = useHistory();
-    useEffect(() => {
+  const history = useHistory();
+  useEffect(() => {
+    if (multiDataSourceEnabled) {
       history.replace({
         search: queryString.stringify({
           dataSourceId,
           dataSourceLabel,
         }),
       });
-    }, [dataSourceId, dataSourceLabel]);
-  }
+    }
+  }, [dataSourceId, dataSourceLabel, multiDataSourceEnabled]);
 }