diff --git a/pyproject.toml b/pyproject.toml
index fca25fb..97f969c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,10 +29,10 @@ classifiers = [
 ]
 dependencies = [
   "capellambse>=0.6.6,<0.7",
-  "capellambse_context_diagrams>=0.4.0",
+  "capellambse_context_diagrams@git+https://github.com/DSD-DBS/capellambse-context-diagrams.git#egg=feat-cache-collection",
   "click",
   "PyYAML",
-  "polarion-rest-api-client==1.2.0",
+  "polarion-rest-api-client@git+https://github.com/DSD-DBS/polarion-rest-api-client.git#egg=fix-description",
   "bidict",
   "cairosvg",
   "jinja2",
diff --git a/tests/test_elements.py b/tests/test_elements.py
index 43377df..ed297ec 100644
--- a/tests/test_elements.py
+++ b/tests/test_elements.py
@@ -889,7 +889,7 @@ def test_update_deleted_work_item(
     def test_create_new_work_item(base_object: BaseObjectContainer):
         polarion_api_get_all_work_items = mock.MagicMock()
         polarion_api_get_all_work_items.return_value = [
-            data_models.CapellaWorkItem(
+            data_model.CapellaWorkItem(
                 id="Obj-1",
                 type="type",
                 uuid_capella="uuid1",
@@ -913,7 +913,7 @@ def test_create_new_work_item(base_object: BaseObjectContainer):
         assert polarion_api_create_work_items.call_count == 1
         assert len(polarion_api_create_work_items.call_args[0][0]) == 1
         work_item = polarion_api_create_work_items.call_args[0][0][0]
-        assert isinstance(work_item, data_models.CapellaWorkItem)
+        assert isinstance(work_item, data_model.CapellaWorkItem)
         assert work_item.id == "AUTO-0"
         assert len(base_object.pw.polarion_data_repo) == 2
         assert (