diff --git a/tests/integration/functional/artifacts/test_artifacts_linage.py b/tests/integration/functional/artifacts/test_artifacts_linage.py
index 3536513dcb..02b33d1df0 100644
--- a/tests/integration/functional/artifacts/test_artifacts_linage.py
+++ b/tests/integration/functional/artifacts/test_artifacts_linage.py
@@ -319,25 +319,25 @@ def my_pipeline():
         prr: PipelineRunResponse = my_pipeline.with_options(
             enable_cache=cache
         )()
-        assert len(prr.steps["producer_step"].input_types) == 0
+        assert len(prr.steps["producer_step"].inputs) == 0
         assert (
-            prr.steps["cs1"].input_types["shared_name"]
+            prr.steps["cs1"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.STEP_OUTPUT
         )
         assert (
-            prr.steps["cs2"].input_types["shared_name"]
+            prr.steps["cs2"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.EXTERNAL
         )
         assert (
-            prr.steps["cs3"].input_types["shared_name"]
+            prr.steps["cs3"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.LAZY_LOADED
         )
         assert (
-            prr.steps["mcsl"].input_types["shared_name"]
+            prr.steps["mcsl"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.MANUAL
         )
         assert (
-            prr.steps["mcsc"].input_types["shared_name"]
+            prr.steps["mcsc"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.MANUAL
         )