diff --git a/services/web/public/coffee/ide/editor/directives/aceEditor/undo/UndoManager.coffee b/services/web/public/coffee/ide/editor/directives/aceEditor/undo/UndoManager.coffee
index 918413d609..1f53fffe52 100644
--- a/services/web/public/coffee/ide/editor/directives/aceEditor/undo/UndoManager.coffee
+++ b/services/web/public/coffee/ide/editor/directives/aceEditor/undo/UndoManager.coffee
@@ -14,14 +14,9 @@ define [
 
 			@editor.on "changeSession", (e) =>
 				@reset()
-				# @session = e.session
+				@session = e.session
 				e.session.setUndoManager(@)
 
-		addSession: (session) ->
-			@session = session
-
-		addSelection: () ->
-
 		showUndoConflictWarning: () ->
 			@$scope.$apply () =>
 				@$scope.undo.show_remote_warning = true
@@ -36,8 +31,7 @@ define [
 			@undoStack = []
 			@redoStack = []
 
-		add: (delta, allowMerge, session) ->
-			return
+		execute: (options) ->
 			if @firstUpdate
 				# The first update we receive is Ace setting the document, which we should
 				# ignore