diff --git a/force_bdss/io/workflow_writer.py b/force_bdss/io/workflow_writer.py
index bb8039e66c6a523281ca35e0c1816cb19564b647..a1e6f8212135de20ff8710f769d79e36983dc26e 100644
--- a/force_bdss/io/workflow_writer.py
+++ b/force_bdss/io/workflow_writer.py
@@ -77,5 +77,5 @@ class WorkflowWriter(HasStrictTraits):
 
         return {
             "id": model.factory.id,
-            "model_data": model.__getstate__()
+            "model_data": state
         }