diff --git a/src/py/crankshaft/test/test_clustering_kmeans.py b/src/py/crankshaft/test/test_clustering_kmeans.py
index 5e096ca..c78861d 100644
--- a/src/py/crankshaft/test/test_clustering_kmeans.py
+++ b/src/py/crankshaft/test/test_clustering_kmeans.py
@@ -69,7 +69,7 @@ class KMeansNonspatialTest(unittest.TestCase):
                                  ("rowids", [1, 2, 3, 4, 5, 6])])]
 
         random_seeds.set_random_seeds(1234)
-        kmeans = Kmeans(FakeQueryRunner(data_raw))
+        kmeans = Kmeans(FakeDataProvider(data_raw))
         clusters = kmeans.nonspatial('subquery', ['col1', 'col2'], 2)
         print str([c[0] for c in clusters])