diff --git a/erpnext/patches/v13_0/setup_gratuity_rule_for_india_and_uae.py b/erpnext/patches/v13_0/setup_gratuity_rule_for_india_and_uae.py
index a71f33c..01fd6a1 100644
--- a/erpnext/patches/v13_0/setup_gratuity_rule_for_india_and_uae.py
+++ b/erpnext/patches/v13_0/setup_gratuity_rule_for_india_and_uae.py
@@ -8,9 +8,9 @@
     frappe.reload_doc('payroll', 'doctype', 'gratuity_rule')
     frappe.reload_doc('payroll', 'doctype', 'gratuity_rule_slab')
     frappe.reload_doc('payroll', 'doctype', 'gratuity_applicable_component')
-    if frappe.db.exists("company", {"country": "India"}):
+    if frappe.db.exists("Company", {"country": "India"}):
         from erpnext.regional.india.setup import create_gratuity_rule
         create_gratuity_rule()
-    if frappe.db.exists("company", {"country": "United Arab Emirates"}):
+    if frappe.db.exists("Company", {"country": "United Arab Emirates"}):
         from erpnext.regional.united_arab_emirates.setup import create_gratuity_rule
-        create_gratuity_rule()
\ No newline at end of file
+        create_gratuity_rule()