diff --git a/selfdrive/updated/tests/test_base.py b/selfdrive/updated/tests/test_base.py
index 1107a2d3b1..b59f03fe77 100644
--- a/selfdrive/updated/tests/test_base.py
+++ b/selfdrive/updated/tests/test_base.py
@@ -10,12 +10,15 @@ import unittest
 from unittest import mock
 
 import pytest
+
+from openpilot.common.params import Params
 from openpilot.selfdrive.manager.process import ManagerProcess
+from openpilot.selfdrive.test.helpers import processes_context
 
 
-from openpilot.selfdrive.test.helpers import processes_context
-from openpilot.common.params import Params
-from openpilot.system.updated.common import get_consistent_flag
+def get_consistent_flag(path: str) -> bool:
+  consistent_file = pathlib.Path(os.path.join(path, ".overlay_consistent"))
+  return consistent_file.is_file()
 
 
 def run(args, **kwargs):
diff --git a/selfdrive/updated/updated.py b/selfdrive/updated/updated.py
index 3a710ba02f..c1af41bd92 100755
--- a/selfdrive/updated/updated.py
+++ b/selfdrive/updated/updated.py
@@ -16,9 +16,9 @@ from markdown_it import MarkdownIt
 from openpilot.common.basedir import BASEDIR
 from openpilot.common.params import Params
 from openpilot.common.time import system_time_valid
-from openpilot.system.hardware import AGNOS, HARDWARE
 from openpilot.common.swaglog import cloudlog
 from openpilot.selfdrive.controls.lib.alertmanager import set_offroad_alert
+from openpilot.system.hardware import AGNOS, HARDWARE
 from openpilot.system.version import get_build_metadata
 
 LOCK_FILE = os.getenv("UPDATER_LOCK_FILE", "/tmp/safe_staging_overlay.lock")