diff --git a/tests/test_edocument.py b/tests/test_edocument.py
index 6da2a59..401959f 100644
--- a/tests/test_edocument.py
+++ b/tests/test_edocument.py
@@ -10,11 +10,10 @@ from decimal import Decimal
 from datetime import date
 from trytond.tests.test_tryton import ModuleTestCase, with_transaction
 from trytond.pool import Pool
-from trytond.modules.edocument_uncefact.tests.test_edocument_uncefact import get_invoice
+from trytond.modules.edocument_uncefact.tests.test_edocument_uncefact import \
+    get_invoice
 from trytond.modules.company.tests import create_company, set_company
 from trytond.exceptions import UserError
-from unittest.mock import Mock
-from decimal import Decimal
 
 
 class EdocTestCase(ModuleTestCase):