diff --git a/sloth/annotations/model.py b/sloth/annotations/model.py
index 59fca57..382f12c 100644
--- a/sloth/annotations/model.py
+++ b/sloth/annotations/model.py
@@ -335,7 +335,7 @@ class AnnotationModelItem(KeyValueModelItem):
     def data(self, role=Qt.DisplayRole, column=0):
         if role == Qt.DisplayRole:
             if column == 0:
-                return self['type']
+                return self['class']
             else:
                 return ""
         elif role == DataRole: