1 files changed, 3 insertions, 2 deletions
diff --git a/src/mongo/db/exec/update.h b/src/mongo/db/exec/update.h
index 4d59dca50f2..c728bda5392 100644
--- a/src/mongo/db/exec/update.h
+++ b/src/mongo/db/exec/update.h
@@ -84,7 +84,8 @@ public:
     virtual bool isEOF();
     virtual StageState work(WorkingSetID* out);
 
-    virtual void doRestoreState(OperationContext* opCtx);
+    virtual void doRestoreState();
+    virtual void doReattachToOperationContext(OperationContext* opCtx);
 
     virtual StageType stageType() const {
         return STAGE_UPDATE;
@@ -163,7 +164,7 @@ private:
     /**
      * Helper for restoring the state of this update.
      */
-    Status restoreUpdateState(OperationContext* opCtx);
+    Status restoreUpdateState();
 
     // Transactional context.  Not owned by us.
     OperationContext* _txn;