diff --git a/frappe/tests/ui_test_helpers.py b/frappe/tests/ui_test_helpers.py
index aab988b2f4..667bb63b83 100644
--- a/frappe/tests/ui_test_helpers.py
+++ b/frappe/tests/ui_test_helpers.py
@@ -634,7 +634,7 @@ def add_remove_role(action: str, user: str, role: str):
 @whitelist_for_tests()
 def publish_realtime(
 	event: str | None = None,
-	message: dict | None = None,
+	message: str | dict | None = None,
 	room: str | None = None,
 	user: str | None = None,
 	doctype: str | None = None,