diff --git a/pyanaconda/modules/storage/devicetree/viewer.py b/pyanaconda/modules/storage/devicetree/viewer.py
index 182fae53c4b..831351fe65d 100644
--- a/pyanaconda/modules/storage/devicetree/viewer.py
+++ b/pyanaconda/modules/storage/devicetree/viewer.py
@@ -139,9 +139,7 @@ def _set_device_data(self, device, data):
         data.attrs["bus"] = self._get_attribute(device, "bus")
         data.attrs["wwn"] = self._get_attribute(device, "wwn")
         data.attrs["uuid"] = self._get_attribute(device, "uuid")
-
-        if isinstance(device, PartitionDevice):
-            data.attrs["partition-label"] = self._get_attribute(device.parted_partition, "name")
+        data.attrs["partition-label"] = self._get_attribute(device, "part_type_name")
 
     def _set_device_data_dasd(self, device, data):
         """Set data for a DASD device."""