diff --git a/erpnext/patches/edigest_enable_income_year_to_date.py b/erpnext/patches/edigest_enable_income_year_to_date.py
index 16e8932ed3..380de832b8 100644
--- a/erpnext/patches/edigest_enable_income_year_to_date.py
+++ b/erpnext/patches/edigest_enable_income_year_to_date.py
@@ -1,4 +1,5 @@
 import webnotes
+from webnotes.model.doc import Document
 
 def execute():
 	companies_list = webnotes.conn.sql("SELECT company_name FROM `tabCompany`", as_list=1)