diff --git a/tests/test_connection.py b/tests/test_connection.py
index 78f7e52f..57141958 100644
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -272,7 +272,10 @@ class ConnectionTest(unittest.TestCase):
         self.assertIsInstance(conn, mongomock.MongoClient)
 
         disconnect_all()
-        class SomeDocument(Document): pass
+
+        class SomeDocument(Document):
+            pass
+
         conn = connect(host="mongomock://localhost:27017/mongoenginetest8")
         some_document = SomeDocument()
         some_document.save()