diff --git a/erpnext/hooks.py b/erpnext/hooks.py
index b8582a3644b..897d4af37a6 100644
--- a/erpnext/hooks.py
+++ b/erpnext/hooks.py
@@ -72,7 +72,10 @@ doctype_list_js = {
 
 page_js = {"print": "public/js/print.js"}
 
-extend_doctype_class = {"Address": "erpnext.accounts.custom.address.ERPNextAddress"}
+extend_doctype_class = {
+	"Address": "erpnext.accounts.custom.address.ERPNextAddress",
+	"Data Import": "erpnext.stock.serial_batch_import.SerialBatchDataImport",
+}
 
 override_whitelisted_methods = {"frappe.www.contact.send_message": "erpnext.templates.utils.send_message"}
 
@@ -382,9 +385,6 @@ pre_submit_validation_doctypes = [
 	"Sales Order",
 ]
 
-
-extend_doctype_class["Data Import"] = "erpnext.stock.serial_batch_import.SerialBatchDataImport"
-
 doc_events = {
 	"*": {
 		"before_print": "erpnext.stock.serial_batch_display.set_serial_number_labels",