diff --git a/src/decentralizepy/training/ChangeAccumulator.py b/src/decentralizepy/training/ChangeAccumulator.py
index 2e105720ebcf175fc07addadbe36954881bbea7d..f485c8b4a06c72c4f3cb23672b9ca141d7fc8c04 100644
--- a/src/decentralizepy/training/ChangeAccumulator.py
+++ b/src/decentralizepy/training/ChangeAccumulator.py
@@ -132,7 +132,7 @@ class ChangeAccumulator(Training):
         Saves the change and the gradient values for every iteration
 
         """
-        tensors_to_cat = [v.data.flatten() for _, v in self.model.items()]
+        tensors_to_cat = [v.data.flatten() for _, v in self.model.state_dict().items()]
         params = torch.abs(torch.cat(tensors_to_cat, dim=0))
         self.save_vector(params, self.model_val_path)