diff --git a/frappe/model/base_document.py b/frappe/model/base_document.py
index 445b92773f..b63ce9dd44 100644
--- a/frappe/model/base_document.py
+++ b/frappe/model/base_document.py
@@ -34,15 +34,9 @@ def get_controller(doctype):
 		from frappe.model.document import Document
 		from frappe.utils.nestedset import NestedSet
 
-		if frappe.flags.in_migrate or frappe.flags.in_install or frappe.flags.in_patch:
-			module_name, custom = ["Core", False]
-		else:
-			# this could be simplified in PY3.8 using walrus operators
-			result = frappe.db.get_value("DocType", doctype, ("module", "custom"), cache=True)
-			if result:
-				module_name, custom = result
-			else:
-				module_name, custom = ["Core", bool(not frappe.db.exists(doctype))]
+		module_name, custom = frappe.db.get_value(
+			"DocType", doctype, ("module", "custom"), cache=True
+		) or ["Core", False]
 
 		if custom:
 			if frappe.db.field_exists("DocType", "is_tree"):