I think, I’ve found some kind of work around to this problem. Since QTreeView::drawBranches() takes the QModelIndex as an argument, I can subclass this method and then draw the icon inside that, if needed. It still needs a bit of fine-tuning, though.