diff --git a/scripts/config.js b/scripts/config.js
index bd2cf83..d528553 100644
--- a/scripts/config.js
+++ b/scripts/config.js
@@ -136,6 +136,7 @@ config.enableCustomGenerator = false
 
 config.customGenerator = [
 	{
+		dimension: ["minecraft:overworld", "minecraft:nether", "minecraft:the_end"],
 		left_block: ["minecraft:stone"],
 		right_block: ["minecraft:dripstone_block"],
 		under_block: [],
@@ -159,6 +160,7 @@ config.customGenerator = [
 		mobs: []
 	},
 	{
+		dimension: ["minecraft:overworld", "minecraft:nether", "minecraft:the_end"],
 		left_block: ["minecraft:flowing_water", "minecraft:water", "WATERLOGGED"],
 		right_block: ["minecraft:flowing_lava", "minecraft:lava"],
 		under_block: ["minecraft:netherite_block"],
@@ -179,6 +181,7 @@ config.customGenerator = [
 		mobs: []
 	},
 	{
+		dimension: ["minecraft:overworld", "minecraft:nether", "minecraft:the_end"],
 		left_block: ["minecraft:flowing_water", "minecraft:water", "WATERLOGGED"],
 		right_block: ["minecraft:flowing_lava", "minecraft:lava"],
 		under_block: ["minecraft:grass_block"],
diff --git a/scripts/main.js b/scripts/main.js
index 53a42a4..2900402 100644
--- a/scripts/main.js
+++ b/scripts/main.js
@@ -193,6 +193,10 @@ function Generator(generatorBlock, gamemode, tool) {
             !config.customGenerator[customGeneratorID].tools.includes(tool.type.id)
         ) return;
 
+        if (isCustomGenerator && config.customGenerator[customGeneratorID].dimension?.length > 0 &&
+            !config.customGenerator[customGeneratorID].dimension.includes(dimension.id)
+        ) return;
+
         if (blocks.length > 0) {
             let blockChances = 0;
             let selectedBlock = blocks[0].chance;