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;