diff --git a/modules/ui/heightmap-editor.js b/modules/ui/heightmap-editor.js
index b744a972..d645cd3f 100644
--- a/modules/ui/heightmap-editor.js
+++ b/modules/ui/heightmap-editor.js
@@ -125,7 +125,8 @@ function editHeightmap() {
 
     customization = 0;
     customizationMenu.style.display = "none";
-    toolsContent.style.display = "block";
+    if (options.querySelector(".tab > button.active").id === "toolsTab")
+      toolsContent.style.display = "block";
     layersPreset.disabled = false;
     exitCustomization.style.display = "none"; // hide finalize button
     restoreDefaultEvents();