diff --git a/pyproject.toml b/pyproject.toml
index 10e477ca9b..32953bed45 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,10 +26,6 @@ codespell = "^2.2.6"
 semver = "^3.0.2"
 mypy = "^1.14.0"
 mock = "^5.0.0"
-attrs = "^25.0.0"
-pyblish-base = "^1.8.7"
-clique = "^2.0.0"
-opentimelineio = "^0.17.0"
 tomlkit = "^0.13.2"
 requests = "^2.32.3"
 mkdocs-material = "^9.6.7"
@@ -42,6 +38,16 @@ pymdown-extensions = "^10.14.3"
 mike = "^2.1.3"
 mkdocstrings-shell = "^1.0.2"
 
+[tool.poetry.group.test.dependencies]
+attrs = "^25.0.0"
+pyblish-base = "^1.8.7"
+clique = "^2.0.0"
+opentimelineio = "^0.17.0"
+speedcopy = "^2.1"
+qtpy="^2.4.3"
+pyside6 = "^6.5.2"
+pytest-ayon = { git = "https://github.com/ynput/pytest-ayon.git", branch = "chore/align-dependencies" }
+
 
 [tool.ruff]
 # Exclude a variety of commonly ignored directories.
diff --git a/tests/client/ayon_core/plugins/publish/test_integrate_traits.py b/tests/client/ayon_core/plugins/publish/test_integrate_traits.py
index c66dda1630..71fcb928c0 100644
--- a/tests/client/ayon_core/plugins/publish/test_integrate_traits.py
+++ b/tests/client/ayon_core/plugins/publish/test_integrate_traits.py
@@ -76,6 +76,7 @@ def sequence_files(tmp_path_factory: pytest.TempPathFactory) -> list[Path]:
 def mock_context(
         project: pytest_ayon.ProjectInfo,
         single_file: Path,
+        folder_path: Path,
         sequence_files: list[Path]) -> pyblish.api.Context:
     """Return a mock instance.
 
@@ -104,6 +105,10 @@ def mock_context(
     instance = context.create_instance("mock_instance")
     instance.data["source"] = "test_source"
     instance.data["families"] = ["render"]
+
+    parents = project.folder_entity["path"].lstrip("/").split("/")
+    hierarchy = "/".join(parents) if parents else ""
+
     instance.data["anatomyData"] = {
         "project": {
             "name": project.project_name,
@@ -121,6 +126,7 @@ def mock_context(
             "name": project.product.name,
             "type": "test"  # pytest-ayon doesn't return the product type yet
         },
+        "hierarchy": hierarchy,
 
     }
     instance.data["folderEntity"] = project.folder_entity