diff --git a/src/dsdk/model.py b/src/dsdk/model.py
index 146377e..f59ad0b 100644
--- a/src/dsdk/model.py
+++ b/src/dsdk/model.py
@@ -29,7 +29,7 @@ class Model:
     YAML = "!model"
     INIT = dumps(
         {
-            "key": "Model.__init__",
+            "key": "model",
             "name": "%s",
             "path": "%s",
             "version": "%s",