From dd98075277bea8c7a44bd6b50727c3ff0c1e4890 Mon Sep 17 00:00:00 2001
From: JayTheBusinessGoose <jaythebusinessgoose@gmail.com>
Date: Mon, 18 Dec 2023 19:44:04 -0700
Subject: [PATCH] Fix level editor biomes for arena stages in the vanilla
 editor.

---
 src/modlunky2/ui/levels/shared/biomes.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/modlunky2/ui/levels/shared/biomes.py b/src/modlunky2/ui/levels/shared/biomes.py
index 8d86878a6..812cbf680 100644
--- a/src/modlunky2/ui/levels/shared/biomes.py
+++ b/src/modlunky2/ui/levels/shared/biomes.py
@@ -75,6 +75,20 @@ def get_biome_for_level(
             return BIOME.VOLCANA
         elif lvl.startswith("volcano") or lvl.endswith("_volcano.lvl"):
             return BIOME.VOLCANA
+
+        dm_themes = [
+            BIOME.DWELLING,
+            BIOME.JUNGLE,
+            BIOME.VOLCANA,
+            BIOME.TIDE_POOL,
+            BIOME.TEMPLE,
+            BIOME.ICE_CAVES,
+            BIOME.NEO_BABYLON,
+            BIOME.SUNKEN_CITY,
+        ]
+        for x, themeselect in enumerate(dm_themes):
+            if lvl.startswith("dm" + str(x + 1)):
+                return themeselect
         return BIOME.DWELLING
 
     # Used for selecting a theme to get the theme code that