diff --git a/src/components/ContentHighlights/tests/ContentHighlights.test.jsx b/src/components/ContentHighlights/tests/ContentHighlights.test.jsx
index 6fc9ca0b38..1561513cb2 100644
--- a/src/components/ContentHighlights/tests/ContentHighlights.test.jsx
+++ b/src/components/ContentHighlights/tests/ContentHighlights.test.jsx
@@ -5,8 +5,8 @@ import { Provider } from 'react-redux';
 import configureMockStore from 'redux-mock-store';
 import thunk from 'redux-thunk';
 import { renderWithRouter } from '@edx/frontend-enterprise-utils';
-import { useHistory } from 'react-router';
 import { IntlProvider } from '@edx/frontend-platform/i18n';
+import { useHistory } from 'react-router-dom';
 import ContentHighlights from '../ContentHighlights';
 import { EnterpriseAppContext } from '../../EnterpriseApp/EnterpriseAppContextProvider';
 
diff --git a/src/components/learner-credit-management/BudgetDetailCatalogTabContents.jsx b/src/components/learner-credit-management/BudgetDetailCatalogTabContents.jsx
index af044e0a0f..297dd18eef 100644
--- a/src/components/learner-credit-management/BudgetDetailCatalogTabContents.jsx
+++ b/src/components/learner-credit-management/BudgetDetailCatalogTabContents.jsx
@@ -4,7 +4,7 @@ import algoliasearch from 'algoliasearch/lite';
 import { Row, Col } from '@edx/paragon';
 
 import { SearchData, SEARCH_FACET_FILTERS } from '@edx/frontend-enterprise-catalog-search';
-import { useHistory } from 'react-router';
+import { useLocation, useHistory } from 'react-router-dom';
 import CatalogSearch from './search/CatalogSearch';
 import {
   LANGUAGE_REFINEMENT,
@@ -14,7 +14,7 @@ import { configuration } from '../../config';
 
 const BudgetDetailCatalogTabContents = () => {
   const history = useHistory();
-  const { location } = history;
+  const location = useLocation();
   const { state: locationState } = location;
   const catalogContainerRef = useRef();