diff --git a/hooks/post_gen_project.py b/hooks/post_gen_project.py
index deb392534..c015bc3be 100644
--- a/hooks/post_gen_project.py
+++ b/hooks/post_gen_project.py
@@ -105,8 +105,12 @@ def remove_async_files():
     file_names = [
         os.path.join("config", "asgi.py"),
         os.path.join("{{cookiecutter.project_slug}}", "users", "websocket.py"),
-        os.path.join("{{cookiecutter.project_slug}}", "users", "tests", "async_server.py"),
-        os.path.join("{{cookiecutter.project_slug}}", "users", "tests", "test_socket.py"),
+        os.path.join(
+            "{{cookiecutter.project_slug}}", "users", "tests", "async_server.py"
+        ),
+        os.path.join(
+            "{{cookiecutter.project_slug}}", "users", "tests", "test_socket.py"
+        ),
     ]
     for file_name in file_names:
         os.remove(file_name)
diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py
index 242402029..93d23e4d5 100644
--- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py
+++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py
@@ -4,15 +4,17 @@ https://github.com/encode/uvicorn/blob/master/tests/protocols/test_websocket.py
 
 """
 from asyncio import new_event_loop
+
 from websockets import connect
 
-from {{ cookiecutter.project_slug }}.users.tests.async_server import run_server
+from {{cookiecutter.project_slug}}.users.tests.async_server import run_server
 
 
 def test_accept_connection():
     """
     If you want to communicate over HTTP, add live_server fixture
     """
+
     async def open_connection(url):
         async with connect(url) as websocket:
             return websocket.open