[FLINK-27214] Fix rescaling benchmark build failed
diff --git a/src/main/java/org/apache/flink/state/benchmark/HashMapStateBackendRescalingBenchmarkExecutor.java b/src/main/java/org/apache/flink/state/benchmark/HashMapStateBackendRescalingBenchmarkExecutor.java
index 33ee9b3..a002e15 100644
--- a/src/main/java/org/apache/flink/state/benchmark/HashMapStateBackendRescalingBenchmarkExecutor.java
+++ b/src/main/java/org/apache/flink/state/benchmark/HashMapStateBackendRescalingBenchmarkExecutor.java
@@ -67,7 +67,7 @@
         benchmark.setUp();
     }
 
-    @Setup(Level.Iteration)
+    @Setup(Level.Invocation)
     public void setUpPerInvocation() throws Exception {
         benchmark.prepareStateForOperator(rescaleType.getSubtaskIndex());
     }
@@ -82,7 +82,7 @@
         benchmark.rescale();
     }
 
-    @TearDown(Level.Iteration)
+    @TearDown(Level.Invocation)
     public void tearDownPerInvocation() throws Exception {
         benchmark.closeOperator();
     }
diff --git a/src/main/java/org/apache/flink/state/benchmark/RocksdbStateBackendRescalingBenchmarkExecutor.java b/src/main/java/org/apache/flink/state/benchmark/RocksdbStateBackendRescalingBenchmarkExecutor.java
index 99a5118..16f18a9 100644
--- a/src/main/java/org/apache/flink/state/benchmark/RocksdbStateBackendRescalingBenchmarkExecutor.java
+++ b/src/main/java/org/apache/flink/state/benchmark/RocksdbStateBackendRescalingBenchmarkExecutor.java
@@ -66,7 +66,7 @@
         benchmark.setUp();
     }
 
-    @Setup(Level.Iteration)
+    @Setup(Level.Invocation)
     public void setUpPerInvocation() throws Exception {
         benchmark.prepareStateForOperator(rescaleType.getSubtaskIndex());
     }
@@ -81,7 +81,7 @@
         benchmark.rescale();
     }
 
-    @TearDown(Level.Iteration)
+    @TearDown(Level.Invocation)
     public void tearDownPerInvocation() throws Exception {
         benchmark.closeOperator();
     }