diff --git a/argilla/tests/integration/test_export_dataset.py b/argilla/tests/integration/test_export_dataset.py
index 5cad1ac996..940ff3c6b5 100644
--- a/argilla/tests/integration/test_export_dataset.py
+++ b/argilla/tests/integration/test_export_dataset.py
@@ -22,6 +22,9 @@
 
 import argilla as rg
 import pytest
+from huggingface_hub.utils._errors import BadRequestError, FileMetadataError, HfHubHTTPError
+
+_RETRIES = 5
 
 
 @pytest.fixture
@@ -71,7 +74,7 @@ def token():
     return os.getenv("HF_TOKEN_ARGILLA_INTERNAL_TESTING")
 
 
-@pytest.mark.flaky(retries=3, only_on=[OSError])  # I/O error hub consistency CICD pipline
+@pytest.mark.flaky(retries=_RETRIES, only_on=[OSError])  # I/O consistency CICD pipline
 @pytest.mark.parametrize("with_records_export", [True, False])
 class TestDiskImportExportMixin:
     def test_export_dataset_to_disk(
@@ -135,6 +138,9 @@ def test_import_dataset_from_disk(
         assert new_dataset.settings.questions[0].name == "label"
 
 
+@pytest.mark.flaky(
+    retries=_RETRIES, only_on=[BadRequestError, FileMetadataError, HfHubHTTPError]
+)  # Hub consistency CICD pipline
 @pytest.mark.skipif(
     not os.getenv("HF_TOKEN_ARGILLA_INTERNAL_TESTING"),
     reason="You are missing a token to write to `argilla-internal-testing` org on the Hugging Face Hub",