diff --git a/nuclia_e2e/nuclia_e2e/tests/nua/test_da_tasks.py b/nuclia_e2e/nuclia_e2e/tests/nua/test_da_tasks.py
index 72a8f86..0e08340 100644
--- a/nuclia_e2e/nuclia_e2e/tests/nua/test_da_tasks.py
+++ b/nuclia_e2e/nuclia_e2e/tests/nua/test_da_tasks.py
@@ -161,7 +161,7 @@ async def wait_for_task_completion(
     client: aiohttp.ClientSession,
     dataset_id: str,
     task_id: str,
-    max_duration: int = 300,
+    max_duration: int = 60 * 8,  # 8 minutes
 ):
     start_time = asyncio.get_event_loop().time()
     while True:
diff --git a/nuclia_e2e/nuclia_e2e/tests/test_kb_features.py b/nuclia_e2e/nuclia_e2e/tests/test_kb_features.py
index 80e2669..82201ae 100644
--- a/nuclia_e2e/nuclia_e2e/tests/test_kb_features.py
+++ b/nuclia_e2e/nuclia_e2e/tests/test_kb_features.py
@@ -49,6 +49,7 @@
 import asyncio
 import backoff
 import base64
+import httpx
 import pytest
 
 Logger = Callable[[str], None]
@@ -95,6 +96,7 @@ async def condition() -> tuple[bool, Any]:
     assert success, "File was not indexed in time, not enough paragraphs found on resource"
 
 
+@backoff.on_exception(backoff.constant, (httpx.ReadError, ClientError), max_tries=3, interval=5)
 async def run_test_import_kb(regional_api_config, ndb: AsyncNucliaDBClient, logger: Logger):
     """
     Imports a kb with three resources and some labelsets already created