From 3703fdb859d0a96da4c6ae5a2b1839c8b433d29a Mon Sep 17 00:00:00 2001
From: Maxime Desroches <desroches.maxime@gmail.com>
Date: Sat, 9 Aug 2025 01:19:08 -0700
Subject: [PATCH] check

---
 system/ui/setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/system/ui/setup.py b/system/ui/setup.py
index 0b679217b3..a0eccf98f8 100755
--- a/system/ui/setup.py
+++ b/system/ui/setup.py
@@ -12,6 +12,7 @@ import shutil
 import pyray as rl
 
 from cereal import log
+from openpilot.common.git import get_short_branch
 from openpilot.system.hardware import HARDWARE
 from openpilot.system.ui.lib.application import gui_app, FontWeight
 from openpilot.system.ui.widgets import Widget
@@ -311,7 +312,7 @@ class Setup(Widget):
     gui_app.set_modal_overlay(self.keyboard, callback=handle_keyboard_result)
 
   def use_openpilot(self):
-    if os.path.isdir(OPENPILOT_CACHE):
+    if os.path.isdir(OPENPILOT_CACHE) and get_short_branch(OPENPILOT_CACHE) == "release3-staging":
       shutil.copyfile(INSTALLER_SOURCE, INSTALLER_DESTINATION)
       self.download_url = OPENPILOT_URL
       self.prepare_installer()