diff --git a/core/src/mindustry/maps/generators/BaseGenerator.java b/core/src/mindustry/maps/generators/BaseGenerator.java
index 5ef5199386..3cd472c4b6 100644
--- a/core/src/mindustry/maps/generators/BaseGenerator.java
+++ b/core/src/mindustry/maps/generators/BaseGenerator.java
@@ -31,7 +31,7 @@ public class BaseGenerator{
     private Seq<Tile> cores;
 
     public static Block getDifficultyWall(int size, float difficulty){
-        Seq<Block> wallsSmall = content.blocks().select(b -> b instanceof Wall && b.size == size
+        Seq<Block> wallsSmall = content.blocks().select(b -> b instanceof Wall && b.isVanilla() && b.size == size
             && !b.insulated && b.buildVisibility == BuildVisibility.shown
             && !(b instanceof Door)
             && !(Structs.contains(b.requirements, i -> state.rules.hiddenBuildItems.contains(i.item))));