diff --git a/python/examples/kmeans.py b/python/examples/kmeans.py
index ad2be2117891b372808c023bc154b979067dd2ff..72cf9f88c6e63bcfb8174251c23d695a215d990c 100644
--- a/python/examples/kmeans.py
+++ b/python/examples/kmeans.py
@@ -33,7 +33,9 @@ if __name__ == "__main__":
     K = int(sys.argv[3])
     convergeDist = float(sys.argv[4])
 
-    kPoints = data.takeSample(False, K, 34)
+    # TODO: change this after we port takeSample()
+    #kPoints = data.takeSample(False, K, 34)
+    kPoints = data.take(K)
     tempDist = 1.0
 
     while tempDist > convergeDist: