diff --git a/tests/integration/vector_io/test_openai_vector_stores.py b/tests/integration/vector_io/test_openai_vector_stores.py
index 8d1f8ab25..457200111 100644
--- a/tests/integration/vector_io/test_openai_vector_stores.py
+++ b/tests/integration/vector_io/test_openai_vector_stores.py
@@ -1118,14 +1118,14 @@ def test_openai_vector_store_file_batch_error_handling(compat_client_with_empty_
     # Invalid files should be marked as failed
     assert batch.file_counts.failed >= 0  # Implementation may vary
     # Test retrieving non-existent batch
-    with pytest.raises((BadRequestError, OpenAIBadRequestError)):  # Should raise an error for non-existent batch
+    with pytest.raises(ValueError):  # Should raise an error for non-existent batch
         compat_client.vector_stores.file_batches.retrieve(
             vector_store_id=vector_store.id,
             batch_id="non_existent_batch_id",
         )
 
     # Test operations on non-existent vector store
-    with pytest.raises((BadRequestError, OpenAIBadRequestError)):  # Should raise an error for non-existent vector store
+    with pytest.raises(ValueError):  # Should raise an error for non-existent vector store
         compat_client.vector_stores.file_batches.create(
             vector_store_id="non_existent_vector_store",
             file_ids=["any_file_id"],