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']