diff --git a/worldedit-core/src/main/java/com/sk89q/worldedit/function/generator/FloraGenerator.java b/worldedit-core/src/main/java/com/sk89q/worldedit/function/generator/FloraGenerator.java
index a25824ca74..b827221532 100644
--- a/worldedit-core/src/main/java/com/sk89q/worldedit/function/generator/FloraGenerator.java
+++ b/worldedit-core/src/main/java/com/sk89q/worldedit/function/generator/FloraGenerator.java
@@ -84,6 +84,8 @@ public void setBiomeAware(boolean biomeAware) {
     public static Pattern getDesertPattern() {
         RandomPattern pattern = new RandomPattern();
         pattern.add(BlockTypes.DEAD_BUSH.getDefaultState(), 30);
+        pattern.add(BlockTypes.SHORT_DRY_GRASS.getDefaultState(), 30);
+        pattern.add(BlockTypes.TALL_DRY_GRASS.getDefaultState(), 20);
         pattern.add(BlockTypes.CACTUS.getDefaultState(), 20);
         pattern.add(BlockTypes.AIR.getDefaultState(), 300);
         return pattern;
@@ -104,6 +106,7 @@ public static Pattern getTemperatePattern() {
             grass = deprecatedGrass;
         }
         pattern.add(grass.getDefaultState(), 300);
+        pattern.add(BlockTypes.BUSH.getDefaultState(), 100);
         pattern.add(BlockTypes.POPPY.getDefaultState(), 5);
         pattern.add(BlockTypes.DANDELION.getDefaultState(), 5);
         return pattern;
@@ -116,7 +119,7 @@ public boolean apply(BlockVector3 position) throws WorldEditException {
         if (block.getBlockType() == BlockTypes.GRASS_BLOCK) {
             editSession.setBlock(position.add(0, 1, 0), temperatePattern.applyBlock(position));
             return true;
-        } else if (block.getBlockType() == BlockTypes.SAND) {
+        } else if (block.getBlockType() == BlockTypes.SAND || BlockCategories.BADLANDS_TERRACOTTA.contains(block.getBlockType())) {
             editSession.setBlock(position.add(0, 1, 0), desertPattern.applyBlock(position));
             return true;
         }