diff --git a/lib/code_update_utils.py b/lib/code_update_utils.py
index dd1cdc3..6b6254b 100644
--- a/lib/code_update_utils.py
+++ b/lib/code_update_utils.py
@@ -65,7 +65,8 @@
     for image_name in images:
         _, image_properties = keyword.run_key(
                 "Get Host Software Property  " + image_name)
-        if image_properties['Purpose'] != var.VERSION_PURPOSE_HOST \
+        if 'Purpose' in image_properties and 'Version' in image_properties \
+                and image_properties['Purpose'] != var.VERSION_PURPOSE_HOST \
                 and image_properties['Version'] != cur_img_version:
             return image_name
     BuiltIn().fail("Did not find any non-running BMC images.")