diff --git a/bin/plot-count-matrix.py b/bin/plot-count-matrix.py
index cdb56c2..8fd9a12 100644
--- a/bin/plot-count-matrix.py
+++ b/bin/plot-count-matrix.py
@@ -88,7 +88,7 @@ ax.set_xticks(np.arange(len(all_classes)))
 ax.set_yticks(np.arange(len(all_clusters)))
 
 ax.set_xticklabels(all_classes)
-ax.set_yticklabels(all_clusters)
+ax.set_yticklabels(list(np.arange(len(all_clusters))))
 
 fig.colorbar(im, ax=ax)