diff --git a/examples/C/mpc/benchmarks/2pc_kmeans.c b/examples/C/mpc/benchmarks/2pc_kmeans.c
index 726f4c91..d4dbc4fa 100644
--- a/examples/C/mpc/benchmarks/2pc_kmeans.c
+++ b/examples/C/mpc/benchmarks/2pc_kmeans.c
@@ -1,7 +1,7 @@
 int main(__attribute__((private(0))) int a[200], __attribute__((private(1))) int b[200]) { 
     int D = 2;
-    int NA = 100;
-    int NB = 100;
+    int NA = 50;
+    int NB = 50;
     int NC = 5;
     int PRECISION = 4;
     int LEN = NA + NB;
@@ -151,8 +151,8 @@ int main(__attribute__((private(0))) int a[200], __attribute__((private(1))) int
         // Compute mean
         for(int i_15 = 0; i_15 < NC; i_15++) {  
             if (count[i_15] > 0) {
-                new_cluster[i_15*D] = new_cluster[i_15*D] / count[i_15];
-                new_cluster[i_15*D+1] = new_cluster[i_15*D] / count[i_15];
+                new_cluster[i_15*D] /= count[i_15];
+                new_cluster[i_15*D+1] /= count[i_15];
             } 
         }
         // ======================= iteration_unrolled_outer
diff --git a/scripts/aby_tests/test_suite.py b/scripts/aby_tests/test_suite.py
index 45f5a25f..22b355dd 100644
--- a/scripts/aby_tests/test_suite.py
+++ b/scripts/aby_tests/test_suite.py
@@ -690,10 +690,10 @@ misc_tests = [
 kmeans_tests = [
     [
         "kmeans",
-        499,
+        283,
         "./third_party/ABY/build/bin/2pc_kmeans",
-        {"a": [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,66,67,68,69,70,71,72,73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,91,92,93,94,95,96,97,98,99,100,101,102,103,104,105,106,107,108,109,110,111,112,113,114,115,116,117,118,119,120,121,122,123,124,125,126,127,128,129,130,131,132,133,134,135,136,137,138,139,140,141,142,143,144,145,146,147,148,149,150,151,152,153,154,155,156,157,158,159,160,161,162,163,164,165,166,167,168,169,170,171,172,173,174,175,176,177,178,179,180,181,182,183,184,185,186,187,188,189,190,191,192,193,194,195,196,197,198,199], "b": 0},
-        {"a": 0, "b": [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,66,67,68,69,70,71,72,73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,91,92,93,94,95,96,97,98,99,100,101,102,103,104,105,106,107,108,109,110,111,112,113,114,115,116,117,118,119,120,121,122,123,124,125,126,127,128,129,130,131,132,133,134,135,136,137,138,139,140,141,142,143,144,145,146,147,148,149,150,151,152,153,154,155,156,157,158,159,160,161,162,163,164,165,166,167,168,169,170,171,172,173,174,175,176,177,178,179,180,181,182,183,184,185,186,187,188,189,190,191,192,193,194,195,196,197,198,199]},
+        {"a": [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,66,67,68,69,70,71,72,73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,91,92,93,94,95,96,97,98,99], "b": 0},
+        {"a": 0, "b": [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,66,67,68,69,70,71,72,73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,91,92,93,94,95,96,97,98,99]},
     ], 
     
 ]