diff --git a/src/main/java/yan/lx/bedrockminer/task/TaskSchemeFinder.java b/src/main/java/yan/lx/bedrockminer/task/TaskSchemeFinder.java
index 1954333..c8cd709 100644
--- a/src/main/java/yan/lx/bedrockminer/task/TaskSchemeFinder.java
+++ b/src/main/java/yan/lx/bedrockminer/task/TaskSchemeFinder.java
@@ -121,7 +121,7 @@ public static BlockPos[] findPistonNearbyRedstoneTorch(BlockPos pistonPos, Clien
             for (int i = 0; i < range; i++) {
                 BlockPos pos = pistonPos.offset(direction, i);
                 BlockState blockState = world.getBlockState(pos);
-                if (blockState.isOf(Blocks.REDSTONE_TORCH) || blockState.isOf(Blocks.REDSTONE_WALL_TORCH)) {
+                if (blockState.isOf(Blocks.REDSTONE_BLOCK) || blockState.isOf(Blocks.REDSTONE_WALL_TORCH)) {
                     list.add(pos);
                 }
             }