diff --git a/tensorflow/core/kernels/cwise_ops_test.cc b/tensorflow/core/kernels/cwise_ops_test.cc
index f0a0d7bccbd16aaf7dd74d4074fde7a3fd8c9659..253974ba6883ce74e499cda06033bc37f0a0be79 100644
--- a/tensorflow/core/kernels/cwise_ops_test.cc
+++ b/tensorflow/core/kernels/cwise_ops_test.cc
@@ -184,7 +184,8 @@ Graph* CubeWithMulSquare(int num) {
   void BM_##DEVICE##_Cube_##Impl(::testing::benchmark::State& state) { \
     const int num = state.range(0);                                    \
                                                                        \
-    test::Benchmark(#DEVICE, Impl(num)).Run(state.iterations());       \
+    test::Benchmark(#DEVICE, Impl(num), /*old_benchmark_api*/ false)   \
+        .Run(state);                                                   \
     const int64 tot = static_cast<int64>(state.iterations()) * num;    \
     state.SetItemsProcessed(tot);                                      \
     state.SetBytesProcessed(tot * sizeof(float));                      \