diff --git a/src/decentralizepy/sharing/PartialModel.py b/src/decentralizepy/sharing/PartialModel.py
index c15c34a3c7261b56172ff93a126ef97c5a894102..7f27d4a10f568a6ad4d59e48d9ec025a7044d79d 100644
--- a/src/decentralizepy/sharing/PartialModel.py
+++ b/src/decentralizepy/sharing/PartialModel.py
@@ -60,7 +60,7 @@ class PartialModel(Sharing):
                 shared_params = dict()
                 shared_params["order"] = self.model.state_dict().keys()
                 shapes = dict()
-                for k, v in self.model.state_dict.items():
+                for k, v in self.model.state_dict().items():
                     shapes[k] = v.shape.tolist()
                 shared_params["shapes"] = shapes