1 files changed, 6 insertions, 2 deletions
diff --git a/src/autotype/AutoTypeMatchModel.cpp b/src/autotype/AutoTypeMatchModel.cpp
index 128015898..207d09b02 100644
--- a/src/autotype/AutoTypeMatchModel.cpp
+++ b/src/autotype/AutoTypeMatchModel.cpp
@@ -20,8 +20,12 @@
 
 #include <QFont>
 
+#include "core/Entry.h"
+#include "core/Global.h"
 #include "core/Group.h"
 #include "core/Metadata.h"
+#include "gui/DatabaseIcons.h"
+#include "gui/Icons.h"
 
 AutoTypeMatchModel::AutoTypeMatchModel(QObject* parent)
     : QAbstractTableModel(parent)
@@ -114,11 +118,11 @@ QVariant AutoTypeMatchModel::data(const QModelIndex& index, int role) const
         switch (index.column()) {
         case ParentGroup:
             if (match.first->group()) {
-                return match.first->group()->iconPixmap();
+                return Icons::groupIconPixmap(match.first->group());
             }
             break;
         case Title:
-            return match.first->iconPixmap();
+            return Icons::entryIconPixmap(match.first);
         }
     } else if (role == Qt::FontRole) {
         QFont font;