diff --git a/modules/terraman/editor/terrain_world_editor.cpp b/modules/terraman/editor/terrain_world_editor.cpp
index b9d9b7acd..ccc818b48 100644
--- a/modules/terraman/editor/terrain_world_editor.cpp
+++ b/modules/terraman/editor/terrain_world_editor.cpp
@@ -329,10 +329,10 @@ EditorPlugin::AfterGUIInput TerrainWorldEditor::forward_spatial_input_event(Came
 			if (mb->get_button_index() == BUTTON_LEFT) {
 				switch (_tool_mode) {
 					case TOOL_MODE_ISOLEVEL_BRUSH: {
-						create_undo_point(_current_action, _isolevel_brush_channel, _isolevel_brush_allow_creating_chunks_button);
+						create_undo_point(_current_action, _isolevel_brush_channel, _isolevel_brush_allow_create_chunks);
 					} break;
 					case TOOL_MODE_PAINT_BRUSH: {
-						create_undo_point(_current_action, _paint_brush_channel, _paint_brush_allow_creating_chunks_button);
+						create_undo_point(_current_action, _paint_brush_channel, _paint_brush_allow_create_chunks);
 					} break;
 					case TOOL_MODE_PAINT_PICKER: {
 						Vector3 position;
diff --git a/modules/terraman/editor/terrain_world_editor.h b/modules/terraman/editor/terrain_world_editor.h
index 6279d9559..aa0040927 100644
--- a/modules/terraman/editor/terrain_world_editor.h
+++ b/modules/terraman/editor/terrain_world_editor.h
@@ -82,6 +82,7 @@ protected:
 
 	// Used by UndoRedo
 	void apply_data(const Array &p_data);
+	
 	void do_chunk_added_action(const Array &p_data);
 	void undo_chunk_added_action(const Array &p_data);
 	void do_chunk_removed_action(const Array &p_data);