diff --git a/openpype/tools/utils/assets_widget.py b/openpype/tools/utils/assets_widget.py
index 9beca69f12..3d4efcdd4d 100644
--- a/openpype/tools/utils/assets_widget.py
+++ b/openpype/tools/utils/assets_widget.py
@@ -513,7 +513,7 @@ class AssetModel(QtGui.QStandardItemModel):
                 item.setData(asset_label, ASSET_LABEL_ROLE)
 
             has_children = item.rowCount() > 0
-            icon = get_asset_icon(asset_data, has_children)
+            icon = get_asset_icon(asset_doc, has_children)
             item.setData(icon, QtCore.Qt.DecorationRole)
 
     def _threaded_fetch(self):
diff --git a/openpype/tools/utils/lib.py b/openpype/tools/utils/lib.py
index 00cec20b2a..d66b636b2a 100644
--- a/openpype/tools/utils/lib.py
+++ b/openpype/tools/utils/lib.py
@@ -135,11 +135,9 @@ def get_project_icon(project_doc):
 
 
 def get_asset_icon_name(asset_doc, has_children=True):
-    if asset_doc:
-        asset_data = asset_doc.get("data") or {}
-        icon_name = asset_data.get("icon")
-        if icon_name:
-            return icon_name
+    icon_name = asset_doc["data"].get("icon")
+    if icon_name:
+        return icon_name
 
     if has_children:
         return "folder"
@@ -147,11 +145,9 @@ def get_asset_icon_name(asset_doc, has_children=True):
 
 
 def get_asset_icon_color(asset_doc):
-    if asset_doc:
-        asset_data = asset_doc.get("data") or {}
-        icon_color = asset_data.get("color")
-        if icon_color:
-            return icon_color
+    icon_color = asset_doc["data"].get("color")
+    if icon_color:
+        return icon_color
     return get_default_entity_icon_color()