diff --git a/CHANGELOG.md b/CHANGELOG.md
index 899890a9..07ad548f 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -8,8 +8,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [Unreleased]
 
 ### Changed
- - Define test client (`substratest.client.Client`) as child class of `substra.Client` (#205)
+
+- Define test client (`substratest.client.Client`) as child class of `substra.Client` (#205)
 ([#257](https://github.com/Substra/substra-tests/pull/257))
+- Remove `model` and `models` for input and output identifiers in tests. Replace by `shared` instead. ([#261](https://github.com/Substra/substra-tests/pull/261))
 
 ## [0.41.0] - 2023-06-12
 
diff --git a/tests/workflows/mnist-fedavg/assets/aggregate_function.py b/tests/workflows/mnist-fedavg/assets/aggregate_function.py
index 858bd7c2..8e65ec21 100644
--- a/tests/workflows/mnist-fedavg/assets/aggregate_function.py
+++ b/tests/workflows/mnist-fedavg/assets/aggregate_function.py
@@ -34,7 +34,7 @@ def forward(self, x):
 def aggregate(inputs, outputs, task_properties):
     # get layers
     inmodels = []
-    for m_path in inputs["models"]:
+    for m_path in inputs["shared"]:
         inmodels.append(load_model(m_path))
 
     model = inmodels[0]