Update public/components/MDSEnabledComponent/MDSEnabledComponent.tsx

Co-authored-by: SuZhou-Joe <suzhou@amazon.com> Signed-off-by: Ramakrishna Chilaka <49393831+RamakrishnaChilaka@users.noreply.github.com>
opensearch-project · Mar 21, 2024 · c208fa9 · c208fa9
1 parent 9ef5c90
commit c208fa9
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]);
 }