diff --git a/test.py b/test.py
index a0a2eec2..b0e476e1 100644
--- a/test.py
+++ b/test.py
@@ -24,7 +24,7 @@
     reader = csv.reader(f)
     next(reader)
     for row in reader:
-        names[int(row[0])] = row[5].split(";")[0]
+        names[int(row[0]) - 1] = row[5].split(";")[0]
 
 
 def visualize_result(data, pred, cfg):