diff --git a/src/IrisDatasetTest.cpp b/src/IrisDatasetTest.cpp
index af6ef35d9..1be17c650 100644
--- a/src/IrisDatasetTest.cpp
+++ b/src/IrisDatasetTest.cpp
@@ -63,7 +63,7 @@ bool load_data(int *samples,
     double *p = &((*input)[0]) + i * input_size;
     double *c = &((*iris_class)[0]) + i * number_classes;
     for (int k = 0; k < number_classes; k++) {
-      c[i] = 0.0;
+      c[k] = 0.0;
     }
 
     fgets(line, 1024, in);