diff --git a/src/py/crankshaft/test/test_clustering_kmeans.py b/src/py/crankshaft/test/test_clustering_kmeans.py
index 8ab300d..e130da4 100644
--- a/src/py/crankshaft/test/test_clustering_kmeans.py
+++ b/src/py/crankshaft/test/test_clustering_kmeans.py
@@ -27,7 +27,7 @@ class KMeansTest(unittest.TestCase):
         """
         data = [{'xs': d['xs'],
                  'ys': d['ys'],
-                 'id': d['id']} for d in self.cluster_data]
+                 'ids': d['ids']} for d in self.cluster_data]
 
         plpy._define_result('select', data)
         clusters = cc.kmeans('subquery', 2)