diff --git a/tests/test_cli.py b/tests/test_cli.py
index 9d600d67..77759f59 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -1,10 +1,11 @@
 import os
 import unittest
-from unittest.mock import mock_open, patch
 from io import StringIO
+from unittest.mock import mock_open, patch
 
 from diagrams.cli import run
 
+
 class CliTest(unittest.TestCase):
     def setUp(self):
         self.test_file = "test_diagram.py"
@@ -29,6 +30,7 @@ with Diagram("test_2", show=False, direction="TB"):
                   EC2("worker4"),
                   EC2("työntekijä 4")] >> RDS("events")
 """
+
     def tearDown(self):
         try:
             os.remove("test.png")
@@ -47,7 +49,6 @@ with Diagram("test_2", show=False, direction="TB"):
         except FileNotFoundError:
             pass
 
-
     def test_run_with_multiple_files(self):
 
         multiple_files = ["file1.py", "file2.py"]