diff --git a/src/suite.py b/src/suite.py
index e458516..f892703 100644
--- a/src/suite.py
+++ b/src/suite.py
@@ -401,7 +401,7 @@ class CVSuite:
                 tag_true = [guess[0] for guess in guesses ]
                 tag_predict = [guess[1] for guess in guesses ]
                 
-                labels = [Tree(tag).name for tag in range(0, 7)]
+                labels = [Tree(tag).name for tag in range(0, 8)]
                 
                 sns.heatmap(confusion_matrix(tag_true, tag_predict), xticklabels=labels, yticklabels=labels, ax=graph, annot=True, cbar=False, fmt='g')
                 graph.set_title(self.models[modelnr][0])