diff --git a/frappe/model/base_document.py b/frappe/model/base_document.py
index 1dbe0f1b0b..ceffc69443 100644
--- a/frappe/model/base_document.py
+++ b/frappe/model/base_document.py
@@ -195,6 +195,7 @@ def get_extended_class(base_class, doctype):
 		(*extension_classes, base_class),
 		{
 			"__reduce__": __reduce__,
+			"__module__": base_class.__module__,
 		},
 	)