diff --git a/pyproject.toml b/pyproject.toml
index c558224..005fac4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -91,6 +91,11 @@ test-cov = 'sh scripts/test/unit_cov.sh'
 test-integration = 'sh scripts/test/integration_test.sh'
 static-check = "pre-commit run --all-files"
 
+[tool.pytest.ini_options]
+filterwarnings = ["ignore::DeprecationWarning"]
+minversion = "6.0"
+markers = ["integration"]
+
 ######################################
 # DOCS
 ######################################
diff --git a/ray_provider/hooks/ray.py b/ray_provider/hooks/ray.py
index 7ca6e3f..2000ecf 100644
--- a/ray_provider/hooks/ray.py
+++ b/ray_provider/hooks/ray.py
@@ -339,6 +339,7 @@ def _wait_for_load_balancer(
 
                 if not lb_details:
                     self.log.info("LoadBalancer details not available yet.")
+                    time.sleep(retry_interval)
                     continue
 
                 working_address = self._check_load_balancer_readiness(lb_details)
diff --git a/tests/test_dag_example.py b/tests/test_dag_example.py
index b6990b4..f265425 100644
--- a/tests/test_dag_example.py
+++ b/tests/test_dag_example.py
@@ -64,6 +64,7 @@ def test_dag_runs(setup_airflow_db, dag_id, dag, fileloc):
         return
 
     try:
-        dag.test()
+        dr = dag.test()
+        assert dr.state == "success"
     except Exception as e:
         pytest.fail(f"Error running DAG {dag_id}: {e}")