diff --git a/services/document-updater/test/unit/coffee/DocumentManager/flushDocTests.coffee b/services/document-updater/test/unit/coffee/DocumentManager/flushDocTests.coffee
index 97085d58b3..551b444f7f 100644
--- a/services/document-updater/test/unit/coffee/DocumentManager/flushDocTests.coffee
+++ b/services/document-updater/test/unit/coffee/DocumentManager/flushDocTests.coffee
@@ -24,7 +24,7 @@ describe "DocumentUpdater - flushDocIfLoaded", ->
 	describe "when the doc is in Redis", ->
 		beforeEach ->
 			@RedisManager.getDoc = sinon.stub().callsArgWith(1, null, @lines, @version)
-			@TrackChangesManager.flushDocChangesIfNeeded = sinon.stub().callsArg(2)
+			@TrackChangesManager.flushDocChanges = sinon.stub().callsArg(2)
 			@PersistenceManager.setDoc = sinon.stub().callsArgWith(4)
 			@DocumentManager.flushDocIfLoaded @project_id, @doc_id, @callback