diff --git a/florist/tests/integration/api/test_train.py b/florist/tests/integration/api/test_train.py
index 3215665..288250d 100644
--- a/florist/tests/integration/api/test_train.py
+++ b/florist/tests/integration/api/test_train.py
@@ -64,14 +64,12 @@ def test_train():
                 client_uuid = response.json()["client_uuids"][0]
 
                 # Wait for training to finish
-                wait_for_metric(server_uuid, "fit_end", test_redis_host,
-                                test_redis_port, LOGGER, max_retries=80)
+                wait_for_metric(server_uuid, "fit_end", test_redis_host, test_redis_port, LOGGER, max_retries=80)
 
                 # Check server metrics
                 server_metrics_result = redis_conn.get(server_uuid)
                 assert server_metrics_result is not None and isinstance(server_metrics_result, bytes)
-                server_metrics = json.loads(
-                    server_metrics_result.decode("utf8"))
+                server_metrics = json.loads(server_metrics_result.decode("utf8"))
                 assert server_metrics["type"] == "server"
                 assert "fit_start" in server_metrics
                 assert "fit_end" in server_metrics
diff --git a/florist/tests/unit/api/test_client.py b/florist/tests/unit/api/test_client.py
index 6f4c8d0..5f06bcf 100644
--- a/florist/tests/unit/api/test_client.py
+++ b/florist/tests/unit/api/test_client.py
@@ -2,7 +2,6 @@
 import json
 from unittest.mock import ANY, Mock, patch
 
-from fastapi.responses import JSONResponse
 from florist.api import client
 from florist.api.clients.mnist import MnistClient
 from florist.api.monitoring.logs import get_client_log_file_path
@@ -89,4 +88,4 @@ def test_check_status(mock_redis: Mock) -> None:
 
     response = client.check_status(test_uuid, test_redis_host, test_redis_port)
 
-    assert json.loads(response.body.decode()) == {"info": "test"}
\ No newline at end of file
+    assert json.loads(response.body.decode()) == {"info": "test"}