From 91e4f1cffc8b14c767eb679c0660ebff25e17753 Mon Sep 17 00:00:00 2001
From: Acid Bubbles <acidbubbles.anon@gmail.com>
Date: Tue, 22 Nov 2022 23:13:47 -0500
Subject: [PATCH] When recording float params, return to the Record screen when
 recording is complete

---
 src/UI/Screens/RecordScreen.cs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/UI/Screens/RecordScreen.cs b/src/UI/Screens/RecordScreen.cs
index e4a17071..f9581bfc 100644
--- a/src/UI/Screens/RecordScreen.cs
+++ b/src/UI/Screens/RecordScreen.cs
@@ -194,8 +194,12 @@ private IEnumerator OnRecordCo(int timeMode, bool recordExtendsLength, bool hide
                 showStartMarkers
             );
 
+            var changedScreen = false;
             if (!targets.OfType<FreeControllerV3AnimationTarget>().Any() && targets.OfType<JSONStorableFloatAnimationTarget>().Any())
+            {
                 ChangeScreen(TargetsScreen.ScreenName);
+                changedScreen = true;
+            }
 
             while (enumerator.MoveNext())
                 yield return enumerator.Current;
@@ -210,6 +214,12 @@ private IEnumerator OnRecordCo(int timeMode, bool recordExtendsLength, bool hide
             animationEditContext.ResetScrubberRange();
             if (hideMenuDuringRecording)
                 SuperController.singleton.ShowMainHUDAuto();
+            if (changedScreen)
+            {
+                yield return 0;
+                // Note: The listener has been disposed, we have to ask the plugin
+                plugin.ChangeScreen(ScreenName, null);
+            }
         }
     }
 }