1 files changed, 4 insertions, 0 deletions
diff --git a/src/mongo/db/exec/sbe/sbe_plan_stage_test.cpp b/src/mongo/db/exec/sbe/sbe_plan_stage_test.cpp
index c2312e19fcd..51078c6d442 100644
--- a/src/mongo/db/exec/sbe/sbe_plan_stage_test.cpp
+++ b/src/mongo/db/exec/sbe/sbe_plan_stage_test.cpp
@@ -69,6 +69,10 @@ PlanStageTestFixture::generateVirtualScanMulti(int32_t numSlots, const BSONArray
 
 void PlanStageTestFixture::prepareTree(CompileCtx* ctx, PlanStage* root) {
     Lock::GlobalLock globalLock{operationContext(), MODE_IS};
+    if (_yieldPolicy) {
+        _yieldPolicy->clearRegisteredPlans();
+        _yieldPolicy->registerPlan(root);
+    }
     root->attachToOperationContext(operationContext());
     root->prepare(*ctx);
     root->open(false);