Fix exceptions

networktocode · Dec 15, 2023 · 5d6cb29 · 5d6cb29
1 parent 38a37ab
commit 5d6cb29
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyntc/devices/asa_device.py b/pyntc/devices/asa_device.py
@@ -600,7 +600,7 @@ def install_os(self, image_name, **vendor_specifics):
             self._wait_for_device_reboot(timeout=timeout)
             if not self._image_booted(image_name):
                 log.error("Host %s: OS install error for image %s", self.host, image_name)
-                raise OSInstallError(hostname=self.facts.get("hostname"), desired_boot=image_name)
+                raise OSInstallError(hostname=self.hostname, desired_boot=image_name)
 
             log.info("Host %s: OS image %s installed successfully.", self.host, image_name)
             return True

diff --git a/pyntc/devices/nxos_device.py b/pyntc/devices/nxos_device.py
@@ -305,7 +305,7 @@ def install_os(self, image_name, **vendor_specifics):
             self._wait_for_device_reboot(timeout=timeout)
             if not self._image_booted(image_name):
                 log.error("Host %s: OS install error for image %s", self.host, image_name)
-                raise OSInstallError(hostname=self.facts.get("hostname"), desired_boot=image_name)
+                raise OSInstallError(hostname=self.hostname, desired_boot=image_name)
             self.save()
 
             log.info("Host %s: OS image %s installed successfully.", self.host, image_name)