diff --git a/tools/automation/tests/__init__.py b/tools/automation/tests/__init__.py
index 37975b48b54..1a2a87cc95c 100644
--- a/tools/automation/tests/__init__.py
+++ b/tools/automation/tests/__init__.py
@@ -22,7 +22,7 @@
 
 
 def extract_module_name(path):
-    mod_name_regex = re.compile(r'azure-cli-([^/\\]*)')
+    mod_name_regex = re.compile(r'azure[/\\]cli[/\\]([^/\\]+)')
     ext_name_regex = re.compile(r'.*(azext_[^/\\]+).*')
 
     try: