diff --git a/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/PruneOlapScanTabletTest.java b/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/PruneOlapScanTabletTest.java
index 385657cd3f229e..d7b652edae258a 100644
--- a/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/PruneOlapScanTabletTest.java
+++ b/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/PruneOlapScanTabletTest.java
@@ -47,7 +47,7 @@
 import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Lists;
 import mockit.Expectations;
-import mockit.Mocked;
+import mockit.Injectable;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 
@@ -57,9 +57,9 @@
 class PruneOlapScanTabletTest extends SqlTestBase implements MemoPatternMatchSupported {
 
     @Test
-    void testPruneOlapScanTablet(@Mocked OlapTable olapTable,
-            @Mocked Partition partition, @Mocked MaterializedIndex index,
-            @Mocked HashDistributionInfo distributionInfo) {
+    void testPruneOlapScanTablet(@Injectable OlapTable olapTable,
+            @Injectable Partition partition, @Injectable MaterializedIndex index,
+            @Injectable HashDistributionInfo distributionInfo) {
         List<Long> tabletIds = Lists.newArrayListWithExpectedSize(300);
         for (long i = 0; i < 300; i++) {
             tabletIds.add(i);