diff --git a/docs_src/nosql_databases/tutorial001.py b/docs_src/nosql_databases/tutorial001.py
index f402f7ffa..66369d54e 100644
--- a/docs_src/nosql_databases/tutorial001.py
+++ b/docs_src/nosql_databases/tutorial001.py
@@ -23,10 +23,7 @@ class Task(TaskBase):
 
 class CassandraConnection:
     def __init__(self, hosts=None, port=9042):
-        if hosts is None:
-            hosts = ["cassandra"]
-
-        self.cluster = Cluster(hosts, port=port)
+        self.cluster = Cluster(hosts or ["cassandra"], port=port)
         self.session = None
         self.keyspace = "task_manager"
 
diff --git a/docs_src/nosql_databases/tutorial001_scylla.py b/docs_src/nosql_databases/tutorial001_scylla.py
index c743fa907..cc8808b65 100644
--- a/docs_src/nosql_databases/tutorial001_scylla.py
+++ b/docs_src/nosql_databases/tutorial001_scylla.py
@@ -23,10 +23,7 @@ class Task(TaskBase):
 
 class ScyllaDBConnection:
     def __init__(self, hosts=None, port=9042):
-        if hosts is None:
-            hosts = ["scylladb"]
-
-        self.cluster = Cluster(hosts, port=port)
+        self.cluster = Cluster(hosts or ["scylladb"], port=port)
         self.session = None
         self.keyspace = "task_manager"
 
diff --git a/tests/test_tutorial/test_nosql_databases/test_tutorial001.py b/tests/test_tutorial/test_nosql_databases/test_tutorial001.py
index e76d69c51..99f2d7999 100644
--- a/tests/test_tutorial/test_nosql_databases/test_tutorial001.py
+++ b/tests/test_tutorial/test_nosql_databases/test_tutorial001.py
@@ -191,6 +191,17 @@ def test_crud_app(client: TestClient):
     assert response.status_code == 404, response.text
     assert response.json() == snapshot({"detail": "Task not found"})
 
+    response = client.put(
+        f"/tasks/{task_id}",
+        json={
+            "title": "Updated non-existent task",
+            "description": "This should fail",
+            "status": "pending",
+        },
+    )
+    assert response.status_code == 404, response.text
+    assert response.json() == snapshot({"detail": "Task not found"})
+
 
 def test_openapi_schema(client: TestClient):
     response = client.get("/openapi.json")