From 37da220e18261f31aecc526571368097480d3440 Mon Sep 17 00:00:00 2001
From: Dalton-V <dalton216@gmail.com>
Date: Mon, 14 Sep 2026 21:47:15 -0500
Subject: [PATCH] Run YOLO shouldn't require a specific controller.

---
 SerialPrograms/Source/ML/Programs/ML_RunYOLO.cpp | 4 ++--
 SerialPrograms/Source/ML/Programs/ML_RunYOLO.h   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/SerialPrograms/Source/ML/Programs/ML_RunYOLO.cpp b/SerialPrograms/Source/ML/Programs/ML_RunYOLO.cpp
index 08f8af6eea..be413cfc3d 100644
--- a/SerialPrograms/Source/ML/Programs/ML_RunYOLO.cpp
+++ b/SerialPrograms/Source/ML/Programs/ML_RunYOLO.cpp
@@ -50,11 +50,11 @@ RunYOLO::RunYOLO()
     PA_ADD_OPTION(MODEL_PATH);
 }
 
-void RunYOLO::program(NintendoSwitch::SingleSwitchProgramEnvironment& env, NintendoSwitch::ProControllerContext& context){
+void RunYOLO::program(NintendoSwitch::SingleSwitchProgramEnvironment& env, CancellableScope& scope){
     std::string model_path = MODEL_PATH;
     YOLOv5Watcher watcher(env.console.overlay(), model_path);
 
-    wait_until(env.console, context, WallClock::max(), {watcher});
+    wait_until(env.console, scope, WallClock::max(), {watcher});
 }
 
 
diff --git a/SerialPrograms/Source/ML/Programs/ML_RunYOLO.h b/SerialPrograms/Source/ML/Programs/ML_RunYOLO.h
index b07fcd204d..385d283561 100644
--- a/SerialPrograms/Source/ML/Programs/ML_RunYOLO.h
+++ b/SerialPrograms/Source/ML/Programs/ML_RunYOLO.h
@@ -25,7 +25,7 @@ class RunYOLO : public NintendoSwitch::SingleSwitchProgramInstance{
 public:
     RunYOLO();
 
-    virtual void program(NintendoSwitch::SingleSwitchProgramEnvironment& env, NintendoSwitch::ProControllerContext& context) override;
+    virtual void program(NintendoSwitch::SingleSwitchProgramEnvironment& env, CancellableScope& scope) override;
 
 private:
     PathOption MODEL_PATH;