diff --git a/modules/ui/heightmap-editor.js b/modules/ui/heightmap-editor.js
index 12783cd1..bb2d1bd0 100644
--- a/modules/ui/heightmap-editor.js
+++ b/modules/ui/heightmap-editor.js
@@ -11,10 +11,13 @@ function editHeightmap() {
     <p>If you need to change the coastline and keep the data, you may try the <i>risk</i> edit option. 
     The data will be restored as much as possible, but the coastline change can cause unexpected fluctuations and errors.</p>
 
-    <p>Check out <a href="https://github.com/Azgaar/Fantasy-Map-Generator/wiki/Heightmap-customization" target="_blank">wiki</a> for guidance.</p>`;
+    <p>Check out <a href="https://github.com/Azgaar/Fantasy-Map-Generator/wiki/Heightmap-customization" target="_blank">wiki</a> for guidance.</p>
+    
+    <p>Please save the map before edditing the heightmap!</p>`;
 
     $("#alert").dialog({resizable: false, title: "Edit Heightmap", width: "28em",
       buttons: {
+        Save: function() {saveMap();},
         Erase: function() {enterHeightmapEditMode("erase");},
         Keep: function() {enterHeightmapEditMode("keep");},
         Risk: function() {enterHeightmapEditMode("risk");},