diff --git a/test/testcases/test_http_api/test_chunk_management_within_dataset/test_update_chunk.py b/test/testcases/test_http_api/test_chunk_management_within_dataset/test_update_chunk.py
index 1281df433..a54d0aa10 100644
--- a/test/testcases/test_http_api/test_chunk_management_within_dataset/test_update_chunk.py
+++ b/test/testcases/test_http_api/test_chunk_management_within_dataset/test_update_chunk.py
@@ -245,4 +245,4 @@ class TestUpdatedChunk:
         delete_documents(HttpApiAuth, dataset_id, {"ids": [document_id]})
         res = update_chunk(HttpApiAuth, dataset_id, document_id, chunk_ids[0])
         assert res["code"] == 102
-        assert res["message"] == f"You don't own the document {document_id}."
+        assert res["message"] in [f"You don't own the document {document_id}.", f"Can't find this chunk {chunk_ids[0]}"]
diff --git a/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py b/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py
index 34d73b99e..23124fa17 100644
--- a/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py
+++ b/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py
@@ -151,4 +151,4 @@ class TestUpdatedChunk:
 
         with pytest.raises(Exception) as excinfo:
             chunks[0].update({})
-        assert f"You don't own the document {chunks[0].document_id}" in str(excinfo.value), str(excinfo.value)
+        assert str(excinfo.value) in [f"You don't own the document {chunks[0].document_id}", f"Can't find this chunk {chunks[0].id}"], str(excinfo.value)