diff --git a/entity-registry/src/main/java/com/linkedin/metadata/aspect/patch/builder/GlobalTagsPatchBuilder.java b/entity-registry/src/main/java/com/linkedin/metadata/aspect/patch/builder/GlobalTagsPatchBuilder.java
index ff34b187f6151..bfd10da37bb3f 100644
--- a/entity-registry/src/main/java/com/linkedin/metadata/aspect/patch/builder/GlobalTagsPatchBuilder.java
+++ b/entity-registry/src/main/java/com/linkedin/metadata/aspect/patch/builder/GlobalTagsPatchBuilder.java
@@ -4,7 +4,7 @@
 import static com.linkedin.metadata.Constants.GLOBAL_TAGS_ASPECT_NAME;
 
 import com.fasterxml.jackson.databind.node.ObjectNode;
-import com.linkedin.common.TagUrn;
+import com.linkedin.common.urn.TagUrn;
 import com.linkedin.metadata.aspect.patch.PatchOperationType;
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
diff --git a/metadata-ingestion/setup.py b/metadata-ingestion/setup.py
index f8d51997330a9..d4e2ada1fc68f 100644
--- a/metadata-ingestion/setup.py
+++ b/metadata-ingestion/setup.py
@@ -468,7 +468,7 @@
     pytest_dep,
     "pytest-asyncio>=0.16.0",
     "pytest-cov>=2.8.1",
-    "pytest-docker>=1.0.1",
+    "pytest-docker>=1.1.0",
     deepdiff_dep,
     "requests-mock",
     "freezegun",
diff --git a/metadata-ingestion/tests/test_helpers/docker_helpers.py b/metadata-ingestion/tests/test_helpers/docker_helpers.py
index 2eb61068196a2..bacb8d80b9e72 100644
--- a/metadata-ingestion/tests/test_helpers/docker_helpers.py
+++ b/metadata-ingestion/tests/test_helpers/docker_helpers.py
@@ -2,7 +2,7 @@
 import logging
 import os
 import subprocess
-from typing import Callable, Optional, Union
+from typing import Callable, Iterator, List, Optional, Union
 
 import pytest
 import pytest_docker.plugin
@@ -37,9 +37,11 @@ def wait_for_port(
         docker_services.wait_until_responsive(
             timeout=timeout,
             pause=pause,
-            check=checker
-            if checker
-            else lambda: is_responsive(container_name, container_port, hostname),
+            check=(
+                checker
+                if checker
+                else lambda: is_responsive(container_name, container_port, hostname)
+            ),
         )
         logger.info(f"Container {container_name} is ready!")
     finally:
@@ -62,14 +64,16 @@ def docker_compose_runner(
 ):
     @contextlib.contextmanager
     def run(
-        compose_file_path: Union[str, list], key: str, cleanup: bool = True
-    ) -> pytest_docker.plugin.Services:
+        compose_file_path: Union[str, List[str]], key: str, cleanup: bool = True
+    ) -> Iterator[pytest_docker.plugin.Services]:
         with pytest_docker.plugin.get_docker_services(
             docker_compose_command=docker_compose_command,
-            docker_compose_file=compose_file_path,
+            # We can remove the type ignore once this is merged:
+            # https://github.com/avast/pytest-docker/pull/108
+            docker_compose_file=compose_file_path,  # type: ignore
             docker_compose_project_name=f"{docker_compose_project_name}-{key}",
             docker_setup=docker_setup,
-            docker_cleanup=docker_cleanup if cleanup else False,
+            docker_cleanup=docker_cleanup if cleanup else [],
         ) as docker_services:
             yield docker_services