diff --git a/worldedit-core/src/main/java/com/sk89q/worldedit/function/pattern/RandomStatePattern.java b/worldedit-core/src/main/java/com/sk89q/worldedit/function/pattern/RandomStatePattern.java
index f2e2870a9..698111dac 100644
--- a/worldedit-core/src/main/java/com/sk89q/worldedit/function/pattern/RandomStatePattern.java
+++ b/worldedit-core/src/main/java/com/sk89q/worldedit/function/pattern/RandomStatePattern.java
@@ -31,15 +31,15 @@ import java.util.stream.Collectors;
 public class RandomStatePattern implements Pattern {
 
     private final Random rand = new Random();
-    private final List<BaseBlock> blocks;
+    private final List<BlockState> blocks;
 
     public RandomStatePattern(FuzzyBlockState state) {
         blocks = state.getBlockType().getAllStates().stream().filter(state::equalsFuzzy)
-                .map(BlockState::toBaseBlock).collect(Collectors.toList());
+                .collect(Collectors.toList());
     }
 
     @Override
     public BaseBlock apply(BlockVector3 position) {
-        return blocks.get(rand.nextInt(blocks.size()));
+        return blocks.get(rand.nextInt(blocks.size())).toBaseBlock();
     }
 }