diff --git a/src/lfx/tests/unit/components/ibm/test_watsonx.py b/src/lfx/tests/unit/components/ibm/test_watsonx.py
index 9678ca70ec..5a4b289a93 100644
--- a/src/lfx/tests/unit/components/ibm/test_watsonx.py
+++ b/src/lfx/tests/unit/components/ibm/test_watsonx.py
@@ -66,9 +66,19 @@ class TestWatsonxAIComponent:
         """Test that API URLs are defined."""
         from lfx.components.ibm.watsonx import WatsonxAIComponent
 
+        expected_urls = [
+            "https://us-south.ml.cloud.ibm.com",
+            "https://eu-de.ml.cloud.ibm.com",
+            "https://eu-gb.ml.cloud.ibm.com",
+            "https://au-syd.ml.cloud.ibm.com",
+            "https://jp-tok.ml.cloud.ibm.com",
+            "https://ca-tor.ml.cloud.ibm.com",
+            "https://ap-south-1.aws.wxai.ibm.com",
+        ]
         assert len(WatsonxAIComponent._urls) > 0
-        assert "https://us-south.ml.cloud.ibm.com" in WatsonxAIComponent._urls
-        assert "https://eu-de.ml.cloud.ibm.com" in WatsonxAIComponent._urls
+
+        for url in expected_urls:
+            assert url in WatsonxAIComponent._urls, f"Expected URL {url} not found in WatsonxAIComponent._urls"
 
     def test_inputs_defined(self, wx_component):
         """Test that all required inputs are defined."""
diff --git a/src/lfx/tests/unit/components/ibm/test_watsonx_embeddings.py b/src/lfx/tests/unit/components/ibm/test_watsonx_embeddings.py
index 98a5d9e7ab..81be0f6c66 100644
--- a/src/lfx/tests/unit/components/ibm/test_watsonx_embeddings.py
+++ b/src/lfx/tests/unit/components/ibm/test_watsonx_embeddings.py
@@ -83,13 +83,18 @@ class TestWatsonxEmbeddingsComponent:
         """Test that URL options are defined."""
         url_input = next(inp for inp in wx_embeddings_component.inputs if inp.name == "url")
 
-        assert "https://us-south.ml.cloud.ibm.com" in url_input.options
-        assert "https://eu-de.ml.cloud.ibm.com" in url_input.options
-        assert "https://eu-gb.ml.cloud.ibm.com" in url_input.options
-        assert "https://au-syd.ml.cloud.ibm.com" in url_input.options
-        assert "https://jp-tok.ml.cloud.ibm.com" in url_input.options
-        assert "https://ca-tor.ml.cloud.ibm.com" in url_input.options
-        assert "https://ap-south-1.aws.wxai.ibm.com" in url_input.options
+        expected_urls = [
+            "https://us-south.ml.cloud.ibm.com",
+            "https://eu-de.ml.cloud.ibm.com",
+            "https://eu-gb.ml.cloud.ibm.com",
+            "https://au-syd.ml.cloud.ibm.com",
+            "https://jp-tok.ml.cloud.ibm.com",
+            "https://ca-tor.ml.cloud.ibm.com",
+            "https://ap-south-1.aws.wxai.ibm.com",
+        ]
+
+        for url in expected_urls:
+            assert url in url_input.options, f"Expected URL {url} not found in options"
 
     @patch("lfx.base.models.model_utils.requests.get")
     def test_fetch_models_success(self, mock_get, mock_response):