diff --git a/modules/milvus/testcontainers/milvus/__init__.py b/modules/milvus/testcontainers/milvus/__init__.py
index 2a1534146..8d8bc2618 100644
--- a/modules/milvus/testcontainers/milvus/__init__.py
+++ b/modules/milvus/testcontainers/milvus/__init__.py
@@ -32,7 +32,7 @@ class MilvusContainer(DockerContainer):
 
             >>> from testcontainers.milvus import MilvusContainer
             >>> with MilvusContainer("milvusdb/milvus:v2.4.4") as milvus_container:
-            ...     milvus_container.get_exposed_port(milvus_container.port) in milvus_container.get_connection_url()
+            ...     str(milvus_container.get_exposed_port(milvus_container.port)) in milvus_container.get_connection_url()
             True
     """
 
diff --git a/modules/milvus/tests/test_milvus.py b/modules/milvus/tests/test_milvus.py
index 12887a49b..56972d630 100644
--- a/modules/milvus/tests/test_milvus.py
+++ b/modules/milvus/tests/test_milvus.py
@@ -21,7 +21,7 @@ def test_run_milvus_success(version: str):
         exposed_port = milvus_container.get_exposed_port(milvus_container.port)
         url = milvus_container.get_connection_url()
 
-    assert url and exposed_port in url
+    assert url and str(exposed_port) in url
 
 
 @pytest.mark.parametrize("version", VERSIONS)