1 files changed, 6 insertions, 4 deletions
diff --git a/tests/custodia.py b/tests/custodia.py
index e14f091..f536dc3 100644
--- a/tests/custodia.py
+++ b/tests/custodia.py
@@ -1,13 +1,15 @@
 # Copyright (C) 2015  Custodia Project Contributors - see LICENSE file
 
 from __future__ import absolute_import
-from tests.client import LocalConnection
+
 import json
 import os
 import subprocess
 import time
 import unittest
 
+from tests.client import LocalConnection
+
 
 class CustodiaTests(unittest.TestCase):
 
@@ -25,9 +27,9 @@ class CustodiaTests(unittest.TestCase):
                            'Content-Type': 'application/json'}
 
     @classmethod
-    def tearDownClass(self):
-        self.custodia_process.kill()
-        self.custodia_process.wait()
+    def tearDownClass(cls):
+        cls.custodia_process.kill()
+        cls.custodia_process.wait()
         for fname in ['server_socket', 'secrets.db']:
             try:
                 os.unlink(fname)