diff --git a/src/communication/azext_communication/commands.py b/src/communication/azext_communication/commands.py
index 31c70b01708..8bdaee867ce 100644
--- a/src/communication/azext_communication/commands.py
+++ b/src/communication/azext_communication/commands.py
@@ -8,7 +8,7 @@
 # pylint: disable=too-many-lines
 # pylint: disable=too-many-statements
 
-# from azure.cli.core.commands import CliCommandType
+from azure.cli.core.commands import CliCommandType
 
 from azext_communication.custom import IdentityAssign
 from azext_communication.custom import IdentityRemove