From 345c37e0205775ec17af41c80dd93bec6d106c3b Mon Sep 17 00:00:00 2001
From: HidyHan <hidyhan5@gmail.com>
Date: Sat, 27 Jan 2024 00:01:39 -0800
Subject: [PATCH] Hardcode the golden data path. (#4588)

---
 tests/ui/ui_tester.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/ui/ui_tester.py b/tests/ui/ui_tester.py
index 2eb668df5..73565b32f 100644
--- a/tests/ui/ui_tester.py
+++ b/tests/ui/ui_tester.py
@@ -306,6 +306,8 @@ def create_path(path):
         return output_dir
 
     def _get_browser_name(self):
+        if self.browser.capabilities['browserName'] == 'chrome-headless-shell':
+            return 'chrome'
         return self.browser.capabilities['browserName']