diff --git a/src/Mod/BIM/bimcommands/BimViews.py b/src/Mod/BIM/bimcommands/BimViews.py index 249eaf1b1b..cd85843547 100644 --- a/src/Mod/BIM/bimcommands/BimViews.py +++ b/src/Mod/BIM/bimcommands/BimViews.py @@ -268,9 +268,8 @@ class BIM_Views: top.setIcon(0, ficon) for v in views: i = QtGui.QTreeWidgetItem([v.Label, ""]) - if hasattr(v.ViewObject, "Proxy"): - if hasattr(v.ViewObject.Proxy, "getIcon"): - i.setIcon(0, QtGui.QIcon(v.ViewObject.Proxy.getIcon())) + if hasattr(v.ViewObject, "Icon"): + i.setIcon(0, v.ViewObject.Icon) i.setToolTip(0, v.Name) top.addChild(i) vm.tree.addTopLevelItem(top) @@ -282,9 +281,8 @@ class BIM_Views: top.setIcon(0, ficon) for p in pages: i = QtGui.QTreeWidgetItem([p.Label, ""]) - if hasattr(v.ViewObject, "Proxy"): - if hasattr(v.ViewObject.Proxy, "getIcon"): - i.setIcon(0, QtGui.QIcon(v.ViewObject.Proxy.getIcon())) + if hasattr(p.ViewObject, "Icon"): + i.setIcon(0, p.ViewObject.Icon) i.setToolTip(0, p.Name) top.addChild(i) vm.tree.addTopLevelItem(top)