diff --git a/src/ai-did-you-mean-this/azext_ai_did_you_mean_this/tests/latest/aladdin_scenario_test_base.py b/src/ai-did-you-mean-this/azext_ai_did_you_mean_this/tests/latest/aladdin_scenario_test_base.py
index ba8215b698e..2003f2e99e5 100644
--- a/src/ai-did-you-mean-this/azext_ai_did_you_mean_this/tests/latest/aladdin_scenario_test_base.py
+++ b/src/ai-did-you-mean-this/azext_ai_did_you_mean_this/tests/latest/aladdin_scenario_test_base.py
@@ -7,7 +7,7 @@
 import logging
 import unittest.mock as mock
 
-from azure_devtools.scenario_tests import mock_in_unit_test
+from azure.cli.testsdk.patches import mock_in_unit_test
 from azure.cli.testsdk import ScenarioTest
 
 from azext_ai_did_you_mean_this._const import UNABLE_TO_HELP_FMT_STR, RECOMMENDATION_HEADER_FMT_STR