diff --git a/tests/test_database.py b/tests/test_database.py
index fd1a59932..8663881a5 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -57,7 +57,8 @@ def test_read_flows(client: TestClient, json_flow: str):
     assert response.json()["name"] == flow.name
     assert response.json()["data"] == flow.data
 
-    flow_style = FlowStyleCreate(color="red", emoji="👍", flow_id=response.json()["id"])
+    flow_style = FlowStyleCreate(
+        color="red", emoji="👍", flow_id=response.json()["id"])
     response = client.post(
         "api/v1/flow_styles/", json=jsonable_encoder(flow_style.dict())
     )
@@ -128,7 +129,8 @@ def test_update_flow(client: TestClient, json_flow: str):
         description="updated description",
         data=data,
     )
-    response = client.patch(f"api/v1/flows/{flow_id}", json=updated_flow.dict())
+    response = client.patch(
+        f"api/v1/flows/{flow_id}", json=updated_flow.dict())
 
     assert response.status_code == 200
     assert response.json()["name"] == updated_flow.name
@@ -237,18 +239,19 @@ def test_create_flow_with_invalid_data(client: TestClient):
 
 def test_get_nonexistent_flow(client: TestClient):
     uuid = uuid4()
-    with pytest.raises(RuntimeError) as excinfo:
-        client.get(f"api/v1/flows/{uuid}")
-    assert str(excinfo.value) == "File at path static/index.html does not exist."
+    response = client.get(f"api/v1/flows/{uuid}")
+    assert response.status_code == 404
 
 
 def test_update_flow_idempotency(client: TestClient, json_flow: str):
     flow_data = json.loads(json_flow)
     data = flow_data["data"]
-    flow_data = FlowCreate(name="Test Flow", description="description", data=data)
+    flow_data = FlowCreate(
+        name="Test Flow", description="description", data=data)
     response = client.post("api/v1/flows/", json=flow_data.dict())
     flow_id = response.json()["id"]
-    updated_flow = FlowCreate(name="Updated Flow", description="description", data=data)
+    updated_flow = FlowCreate(
+        name="Updated Flow", description="description", data=data)
     response1 = client.put(f"api/v1/flows/{flow_id}", json=updated_flow.dict())
     response2 = client.put(f"api/v1/flows/{flow_id}", json=updated_flow.dict())
     assert response1.json() == response2.json()
@@ -263,16 +266,14 @@ def test_update_nonexistent_flow(client: TestClient, json_flow: str):
         description="description",
         data=data,
     )
-    with pytest.raises(RuntimeError) as excinfo:
-        client.patch(f"api/v1/flows/{uuid}", json=updated_flow.dict())
-    assert str(excinfo.value) == "File at path static/index.html does not exist."
+    response = client.patch(f"api/v1/flows/{uuid}", json=updated_flow.dict())
+    assert response.status_code == 404
 
 
 def test_delete_nonexistent_flow(client: TestClient):
     uuid = uuid4()
-    with pytest.raises(RuntimeError) as excinfo:
-        client.delete(f"api/v1/flows/{uuid}")
-    assert str(excinfo.value) == "File at path static/index.html does not exist."
+    response = client.delete(f"api/v1/flows/{uuid}")
+    assert response.status_code == 404
 
 
 def test_read_empty_flows(client: TestClient):
@@ -312,7 +313,8 @@ def test_create_flow_style(client: TestClient):
 def test_read_flow_styles(client: TestClient):
     response = client.get("api/v1/flow_styles/")
     assert response.status_code == 200
-    flow_styles = [FlowStyleRead(**flow_style) for flow_style in response.json()]
+    flow_styles = [FlowStyleRead(**flow_style)
+                   for flow_style in response.json()]
     assert not flow_styles
     # Create test data
     flow_style = FlowStyleCreate(color="red", emoji="🔴")
@@ -321,7 +323,8 @@ def test_read_flow_styles(client: TestClient):
     # Check response data
     response = client.get("api/v1/flow_styles/")
     assert response.status_code == 200
-    flow_styles = [FlowStyleRead(**flow_style) for flow_style in response.json()]
+    flow_styles = [FlowStyleRead(**flow_style)
+                   for flow_style in response.json()]
     assert len(flow_styles) == 1
     assert flow_styles[0].color == flow_style.color
     assert flow_styles[0].emoji == flow_style.emoji
@@ -359,6 +362,5 @@ def test_delete_flow_style(client: TestClient):
     assert response.status_code == 200
     assert response.json() == {"message": "FlowStyle deleted successfully"}
 
-    with pytest.raises(RuntimeError) as excinfo:
-        response = client.get(f"api/v1/flow_styles/{created_flow_style.id}")
-    assert str(excinfo.value) == "File at path static/index.html does not exist."
+    response = client.get(f"api/v1/flow_styles/{created_flow_style.id}")
+    assert response.status_code == 404