diff --git a/addons/website_sale_digital/controllers/main.py b/addons/website_sale_digital/controllers/main.py
index d1a4b777085..c8520e70ac9 100644
--- a/addons/website_sale_digital/controllers/main.py
+++ b/addons/website_sale_digital/controllers/main.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
 import base64
@@ -16,7 +15,7 @@ class WebsiteSaleDigital(CustomerPortal):
 
     @http.route()
     def portal_order_page(self, order_id=None, **post):
-        response = super(WebsiteSaleDigital, self).portal_order_page(order_id=order_id, **post)
+        response = super().portal_order_page(order_id=order_id, **post)
         if not 'sale_order' in response.qcontext:
             return response
         order = response.qcontext['sale_order']
@@ -68,7 +67,7 @@ class WebsiteSaleDigital(CustomerPortal):
         # Check if the user has bought the associated product
         res_model = attachment['res_model']
         res_id = attachment['res_id']
-        purchased_products = request.env['account.move.line'].get_digital_purchases()
+        purchased_products = request.env['account.move.line']._get_digital_purchases()
 
         if res_model == 'product.product':
             if res_id not in purchased_products:
diff --git a/addons/website_sale_digital/models/__init__.py b/addons/website_sale_digital/models/__init__.py
index 4efa7114347..987b812274d 100644
--- a/addons/website_sale_digital/models/__init__.py
+++ b/addons/website_sale_digital/models/__init__.py
@@ -1,6 +1,6 @@
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
-from . import account_invoice
+from . import account_move_line
 from . import ir_attachment
 from . import product_product
 from . import product_template
diff --git a/addons/website_sale_digital/models/account_invoice.py b/addons/website_sale_digital/models/account_move_line.py
similarity index 91%
rename from addons/website_sale_digital/models/account_invoice.py
rename to addons/website_sale_digital/models/account_move_line.py
index a79f3c7cceb..c14e6b5a6b7 100644
--- a/addons/website_sale_digital/models/account_invoice.py
+++ b/addons/website_sale_digital/models/account_move_line.py
@@ -1,14 +1,12 @@
-# -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
 from odoo import models
 
 
-class AccountInvoiceLine(models.Model):
-
+class AccountMoveLine(models.Model):
     _inherit = ['account.move.line']
 
-    def get_digital_purchases(self):
+    def _get_digital_purchases(self):
         partner = self.env.user.partner_id
 
         # Get paid invoices