diff --git a/osu.Game/Screens/Edit/Submission/SubmissionStageProgress.cs b/osu.Game/Screens/Edit/Submission/SubmissionStageProgress.cs
index 49fd28ee62..8af4e3fe52 100644
--- a/osu.Game/Screens/Edit/Submission/SubmissionStageProgress.cs
+++ b/osu.Game/Screens/Edit/Submission/SubmissionStageProgress.cs
@@ -149,9 +149,8 @@ namespace osu.Game.Screens.Edit.Submission
             progress.BindValueChanged(_ => Scheduler.AddOnce(updateProgress), true);
 
             progressSampleChannel = progressSample?.GetChannel();
-            // TODO: add back once framework revert is reverted.
-            // if (progressSampleChannel != null)
-            //     progressSampleChannel.ManualFree = true;
+            if (progressSampleChannel != null)
+                progressSampleChannel.ManualFree = true;
         }
 
         public void SetNotStarted() => status.Value = StageStatusType.NotStarted;