diff --git a/frappe/tests/ui_test_helpers.py b/frappe/tests/ui_test_helpers.py
index c386e65605..f4a26a6325 100644
--- a/frappe/tests/ui_test_helpers.py
+++ b/frappe/tests/ui_test_helpers.py
@@ -7,7 +7,7 @@ UI_TEST_USER = "frappe@example.com"
 
 
 def whitelist_for_tests(fn):
-	if frappe.request and not frappe.flags.in_test and not getattr(frappe.local, "dev_server", 0):
+	if frappe.request and not frappe.flags.in_test and not frappe._dev_server:
 		frappe.throw("Cannot run UI tests. Use a development server with `bench start`")
 
 	return frappe.whitelist()(fn)