diff --git a/services/track-changes/test/unit/coffee/DocArchive/DocArchiveManager.coffee b/services/track-changes/test/unit/coffee/DocArchive/DocArchiveManager.coffee
index f09889492d..fe648b6e67 100644
--- a/services/track-changes/test/unit/coffee/DocArchive/DocArchiveManager.coffee
+++ b/services/track-changes/test/unit/coffee/DocArchive/DocArchiveManager.coffee
@@ -61,8 +61,10 @@ describe "DocArchiveManager", ->
 		beforeEach ->
 			@update = { _id: ObjectId(), op: "op", meta: "meta", v: "v"}
 			@MongoManager.getDocChangesCount = sinon.stub().callsArg(1)
+			@MongoManager.getArchivedDocChanges = sinon.stub().callsArgWith(1, null, 0)
 			@MongoManager.getLastCompressedUpdate = sinon.stub().callsArgWith(1, null, @update)
-			@MongoAWS.archiveDocHistory = sinon.stub().callsArg(2)
+			@MongoAWS.archiveDocHistory = sinon.stub().callsArg(3)
+			@MongoManager.markDocHistoryAsArchiveInProgress = sinon.stub().callsArg(2)
 			@MongoManager.markDocHistoryAsArchived = sinon.stub().callsArg(2)
 			@DocArchiveManager.archiveDocChanges @project_id, @doc_id, @callback
 
diff --git a/services/track-changes/test/unit/coffee/DocArchive/MongoAWS.coffee b/services/track-changes/test/unit/coffee/DocArchive/MongoAWS.coffee
index 0bad498eb5..5667f0fada 100644
--- a/services/track-changes/test/unit/coffee/DocArchive/MongoAWS.coffee
+++ b/services/track-changes/test/unit/coffee/DocArchive/MongoAWS.coffee
@@ -27,6 +27,7 @@ describe "MongoAWS", ->
 
 		@project_id = ObjectId().toString()
 		@doc_id = ObjectId().toString()
+		@update = { v:123 }
 		@callback = sinon.stub()
 		
 	describe "archiveDocHistory", ->
@@ -46,7 +47,7 @@ describe "MongoAWS", ->
 								cb()
 			@JSONStream.stringify = sinon.stub()
 
-			@MongoAWS.archiveDocHistory @project_id, @doc_id, @callback
+			@MongoAWS.archiveDocHistory @project_id, @doc_id, @update, @callback
 
 		it "should call the callback", ->
 			@callback.called.should.equal true
diff --git a/services/track-changes/test/unit/coffee/MongoManager/MongoManagerTests.coffee b/services/track-changes/test/unit/coffee/MongoManager/MongoManagerTests.coffee
index 6c127cd343..5d5f210035 100644
--- a/services/track-changes/test/unit/coffee/MongoManager/MongoManagerTests.coffee
+++ b/services/track-changes/test/unit/coffee/MongoManager/MongoManagerTests.coffee
@@ -446,7 +446,7 @@ describe "MongoManager", ->
 			@db.docHistory.count
 				.calledWith({
 					doc_id: ObjectId(@doc_id)
-					inS3 : true
+					inS3: {$exists: true}
 				}, {
 				})
 				.should.equal true