diff --git a/unified-scheduler-logic/src/lib.rs b/unified-scheduler-logic/src/lib.rs
index b2834e3d627183..8a4c33134dcabb 100644
--- a/unified-scheduler-logic/src/lib.rs
+++ b/unified-scheduler-logic/src/lib.rs
@@ -303,8 +303,9 @@ impl SchedulingStateMachine {
                 self.reschedule_count += 1;
                 self.try_lock_for_task(TaskSource::Retryable, task)
             })
-            .inspect(|_| {
+            .map(|task| {
                 self.rescheduled_task_count += 1;
+                task
             })
     }