diff --git a/plugins/module_utils/image_mgmt/image_upgrade.py b/plugins/module_utils/image_mgmt/image_upgrade.py
index 59223447d..2bdf63116 100644
--- a/plugins/module_utils/image_mgmt/image_upgrade.py
+++ b/plugins/module_utils/image_mgmt/image_upgrade.py
@@ -33,8 +33,6 @@
     ImageInstallOptions
 from ansible_collections.cisco.dcnm.plugins.module_utils.image_mgmt.switch_issu_details import \
     SwitchIssuDetailsByIpAddress
-from ansible_collections.cisco.dcnm.plugins.module_utils.network.dcnm.dcnm import \
-    dcnm_send
 
 
 class ImageUpgrade(ImageUpgradeCommon):
diff --git a/plugins/modules/dcnm_image_upgrade.py b/plugins/modules/dcnm_image_upgrade.py
index 9cb147e50..30358ca22 100644
--- a/plugins/modules/dcnm_image_upgrade.py
+++ b/plugins/modules/dcnm_image_upgrade.py
@@ -436,8 +436,6 @@
     SwitchDetails
 from ansible_collections.cisco.dcnm.plugins.module_utils.image_mgmt.switch_issu_details import \
     SwitchIssuDetailsByIpAddress
-from ansible_collections.cisco.dcnm.plugins.module_utils.network.dcnm.dcnm import \
-    dcnm_send
 
 
 class ImageUpgradeTask(ImageUpgradeCommon):
@@ -1017,9 +1015,13 @@ def _attach_or_detach_image_policy(self, action=None) -> None:
             instance.serial_numbers = value
             instance.commit()
             if action == "attach":
-                self.task_result.response_attach_policy = copy.deepcopy(instance.response_current)
+                self.task_result.response_attach_policy = copy.deepcopy(
+                    instance.response_current
+                )
             if action == "detach":
-                self.task_result.response_detach_policy = copy.deepcopy(instance.response_current)
+                self.task_result.response_detach_policy = copy.deepcopy(
+                    instance.response_current
+                )
 
         for diff in instance.diff:
             msg = (