diff --git a/selfdrive/ui/tests/test_ui/run.py b/selfdrive/ui/tests/test_ui/run.py
index 79f30e79bf..871333563d 100644
--- a/selfdrive/ui/tests/test_ui/run.py
+++ b/selfdrive/ui/tests/test_ui/run.py
@@ -8,9 +8,7 @@ import numpy as np
 import os
 import pywinctl
 import time
-import unittest  # noqa: TID251
 
-from parameterized import parameterized
 from cereal import messaging, car, log
 from cereal.visionipc import VisionIpcServer, VisionStreamType
 
@@ -122,16 +120,11 @@ TEST_OUTPUT_DIR = TEST_DIR / "report"
 SCREENSHOTS_DIR = TEST_OUTPUT_DIR / "screenshots"
 
 
-class TestUI(unittest.TestCase):
-  @classmethod
-  def setUpClass(cls):
+class TestUI:
+  def __init__(self):
     os.environ["SCALE"] = "1"
     sys.modules["mouseinfo"] = False
 
-  @classmethod
-  def tearDownClass(cls):
-    del sys.modules["mouseinfo"]
-
   def setup(self):
     self.sm = SubMaster(["uiDebug"])
     self.pm = PubMaster(["deviceState", "pandaStates", "controlsState", 'roadCameraState', 'wideRoadCameraState', 'liveLocationKalman'])
@@ -147,8 +140,8 @@ class TestUI(unittest.TestCase):
   def screenshot(self):
     import pyautogui
     im = pyautogui.screenshot(region=(self.ui.left, self.ui.top, self.ui.width, self.ui.height))
-    self.assertEqual(im.width, 2160)
-    self.assertEqual(im.height, 1080)
+    assert im.width == 2160
+    assert im.height == 1080
     img = np.array(im)
     im.close()
     return img
@@ -158,7 +151,6 @@ class TestUI(unittest.TestCase):
     pyautogui.click(self.ui.left + x, self.ui.top + y, *args, **kwargs)
     time.sleep(UI_DELAY) # give enough time for the UI to react
 
-  @parameterized.expand(CASES.items())
   @with_processes(["ui"])
   def test_ui(self, name, setup_case):
     self.setup()
@@ -188,7 +180,10 @@ def create_screenshots():
     shutil.rmtree(TEST_OUTPUT_DIR)
 
   SCREENSHOTS_DIR.mkdir(parents=True)
-  unittest.main(exit=False)
+
+  t = TestUI()
+  for name, setup in CASES.items():
+    t.test_ui(name, setup)
 
 if __name__ == "__main__":
   print("creating test screenshots")