diff --git a/frappe/website/doctype/website_settings/website_settings.py b/frappe/website/doctype/website_settings/website_settings.py
index 95e5177282..d0f951e0c4 100644
--- a/frappe/website/doctype/website_settings/website_settings.py
+++ b/frappe/website/doctype/website_settings/website_settings.py
@@ -5,10 +5,10 @@ from __future__ import unicode_literals
 import frappe
 from frappe import _
 from frappe.utils import get_request_site_address, encode
-from frappe.model.controller import DocListController
+from frappe.model.document import Document
 from urllib import quote
 
-class WebsiteSettings(DocListController):
+class WebsiteSettings(Document):
 	def validate(self):
 		self.validate_top_bar_items()
 		self.validate_footer_items()
diff --git a/frappe/website/website_generator.py b/frappe/website/website_generator.py
index b93c9616a8..285be473f5 100644
--- a/frappe/website/website_generator.py
+++ b/frappe/website/website_generator.py
@@ -3,14 +3,14 @@
 
 from __future__ import unicode_literals
 import frappe
-from frappe.model.controller import DocListController
+from frappe.model.document import Document
 from frappe.model.naming import append_number_if_name_exists
 from frappe.website.utils import cleanup_page_name
 from frappe.utils import now
 
 from frappe.website.doctype.website_route.website_route import add_to_sitemap, update_sitemap, remove_sitemap
 
-class WebsiteGenerator(DocListController):
+class WebsiteGenerator(Document):
 	def autoname(self):
 		self.setup_generator()
 		if not self.website_template: return