diff --git a/setup.py b/setup.py
index fa1f0fa..1a484a9 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 """
 
 # Always prefer setuptools over distutils
-from setuptools import setup, find_packages
+from setuptools import setup
 # To use a consistent encoding
 from codecs import open
 from os import path
@@ -36,7 +36,7 @@ with open(path.join(here, 'versiondep.txt'), encoding='utf-8') as f:
         l2 = i.strip().split(';')
         if len(l2) < 4:
             continue
-        modversion[l2[0]] = {'min':l2[1], 'max':l2[2], 'prefix':l2[3]}
+        modversion[l2[0]] = {'min': l2[1], 'max': l2[2], 'prefix': l2[3]}
 
 # tryton-version
 major_version = 6
@@ -51,19 +51,21 @@ for dep in info.get('depends', []):
                 prefix = modversion[dep]['prefix']
 
             if len(modversion[dep]['max']) > 0:
-                requires.append('%s_%s >= %s, <= %s' %
-                    (prefix, dep, modversion[dep]['min'], modversion[dep]['max']))
-            else :
-                requires.append('%s_%s >= %s' %
-                    (prefix, dep, modversion[dep]['min']))
-        else :
-          requires.append('%s_%s >= %s.%s, < %s.%s' %
-                ('trytond', dep, major_version, minor_version,
+                requires.append('%s_%s >= %s, <= %s' % (
+                    prefix, dep, modversion[dep]['min'],
+                    modversion[dep]['max']))
+            else:
+                requires.append('%s_%s >= %s' % (
+                    prefix, dep, modversion[dep]['min']))
+        else:
+            requires.append('%s_%s >= %s.%s, < %s.%s' % (
+                'trytond', dep, major_version, minor_version,
                 major_version, minor_version + 1))
-requires.append('trytond >= %s.%s, < %s.%s' %
-        (major_version, minor_version, major_version, minor_version + 1))
+requires.append('trytond >= %s.%s, < %s.%s' % (
+    major_version, minor_version, major_version, minor_version + 1))
 
-setup(name='%s_%s' % (PREFIX, MODULE),
+setup(
+    name='%s_%s' % (PREFIX, MODULE),
     version=info.get('version', '0.0.1'),
     description='Tryton module to XRechnung to edocument.',
     long_description=long_description,
@@ -72,21 +74,22 @@ setup(name='%s_%s' % (PREFIX, MODULE),
     author_email='service@m-ds.de',
     license='GPL-3',
     classifiers=[
-    'Development Status :: 5 - Production/Stable',
-    'Environment :: Plugins',
-    'Framework :: Tryton',
-    'Intended Audience :: Developers',
-    'Intended Audience :: Customer Service',
-    'Intended Audience :: Information Technology',
-    'Intended Audience :: Financial and Insurance Industry',
-    'Topic :: Office/Business',
-    'Topic :: Office/Business :: Financial :: Accounting',
-    'Natural Language :: German',
-    'Natural Language :: English',
-    'Operating System :: OS Independent',
-    'License :: OSI Approved :: GNU General Public License (GPL)',
-    'Programming Language :: Python :: 3.7',
-    'Programming Language :: Python :: 3.8',
+        'Development Status :: 5 - Production/Stable',
+        'Environment :: Plugins',
+        'Framework :: Tryton',
+        'Intended Audience :: Developers',
+        'Intended Audience :: Customer Service',
+        'Intended Audience :: Information Technology',
+        'Intended Audience :: Financial and Insurance Industry',
+        'Topic :: Office/Business',
+        'Topic :: Office/Business :: Financial :: Accounting',
+        'Natural Language :: German',
+        'Natural Language :: English',
+        'Operating System :: OS Independent',
+        'License :: OSI Approved :: GNU General Public License (GPL)',
+        'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: 3.8',
+        'Programming Language :: Python :: 3.9',
     ],
 
     keywords='tryton xrechnung edcoument',
@@ -95,9 +98,10 @@ setup(name='%s_%s' % (PREFIX, MODULE),
         'trytond.modules.%s' % MODULE,
         ],
     package_data={
-        'trytond.modules.%s' % MODULE: (info.get('xml', [])
+        'trytond.modules.%s' % MODULE: (
+            info.get('xml', [])
             + ['tryton.cfg', 'locale/*.po', 'tests/*.py',
-                'template/*/*.xml','versiondep.txt', 'README.rst']),
+                'template/*/*.xml', 'versiondep.txt', 'README.rst']),
         },
 
     install_requires=requires,
diff --git a/tests/test_edocument.py b/tests/test_edocument.py
index 85c1016..92a1567 100644
--- a/tests/test_edocument.py
+++ b/tests/test_edocument.py
@@ -1,12 +1,12 @@
 # -*- coding: utf-8 -*-
-# This file is part of the cashbook-module from m-ds for Tryton.
+# This file is part of the edocument-module for Tryton from m-ds.de.
 # The COPYRIGHT file at the top level of this repository contains the
 # full copyright notices and license terms.
 
 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 unittest.mock import Mock, MagicMock
+from unittest.mock import Mock
 from decimal import Decimal
 
 
@@ -20,7 +20,6 @@ class EdocTestCase(ModuleTestCase):
         """
         pool = Pool()
         Template = pool.get('edocument.xrechnung.invoice')
-        Address = pool.get('party.address')
         Identifier = pool.get('party.identifier')
         Party = pool.get('party.party')
         Bank = pool.get('bank')
@@ -28,19 +27,22 @@ class EdocTestCase(ModuleTestCase):
         BankNumber = pool.get('bank.account.number')
 
         invoice = get_invoice()
-        invoice.party.get_xrechnung_route_id = Mock(return_value='xrechn-route-id-123')
+        invoice.party.get_xrechnung_route_id = Mock(
+            return_value='xrechn-route-id-123')
         invoice.company.party.bank_accounts = [
-            Mock(spec=BankAccount,
+            Mock(
+                spec=BankAccount,
                 currency=invoice.currency,
                 bank=Mock(spec=Bank, party=Mock(spec=Party, name='Bank')),
-                owners = [invoice.company.party],
-                numbers = [Mock(spec=BankNumber, type='other', number='123456')],
+                owners=[invoice.company.party],
+                numbers=[Mock(spec=BankNumber, type='other', number='123456')],
                 )]
         invoice.description = 'description of invoice'
         invoice.comment = 'note line 1\nnote line 2'
         invoice.taxes[0].tax.rate = Decimal('0.1')
         invoice.identifiers = [
-            Mock(spec=Identifier,
+            Mock(
+                spec=Identifier,
                 type='edoc_route_id',
                 code='xrechn-route-id-123')
             ]
@@ -100,3 +102,6 @@ class EdocTestCase(ModuleTestCase):
             fhdl.write(invoice_string.decode('utf8'))
 
 # end EdocTestCase
+
+
+del ModuleTestCase