diff --git a/modules/ui/heightmap-editor.js b/modules/ui/heightmap-editor.js
index 935660ca..1b4e58bf 100644
--- a/modules/ui/heightmap-editor.js
+++ b/modules/ui/heightmap-editor.js
@@ -405,7 +405,7 @@ function editHeightmap() {
     drawStates();
     drawBorders();
 
-    if (erosion) {
+    if (erosionAllowed) {
       Rivers.specify();
       Lakes.generateName();
     }