diff --git a/src/decentralizepy/sharing/PartialModel.py b/src/decentralizepy/sharing/PartialModel.py
index 49083aaf03957aefe4cfd83c822800dba6758d87..69875ada11a65be9fce5dd17cef8eeeb03b6497c 100644
--- a/src/decentralizepy/sharing/PartialModel.py
+++ b/src/decentralizepy/sharing/PartialModel.py
@@ -58,7 +58,7 @@ class PartialModel(Sharing):
                     shared_params = json.load(inf)
             else:
                 shared_params = dict()
-                shared_params["order"] = self.model.state_dict().keys()
+                shared_params["order"] = list(self.model.state_dict().keys())
                 shapes = dict()
                 for k, v in self.model.state_dict().items():
                     shapes[k] = list(v.shape)