diff --git a/Projects/TensorFlow/Schemas/AppSchema.json b/Projects/TensorFlow/Schemas/AppSchema.json
index c10cfac5d..8cf256384 100644
--- a/Projects/TensorFlow/Schemas/AppSchema.json
+++ b/Projects/TensorFlow/Schemas/AppSchema.json
@@ -455,7 +455,7 @@
         "attachingRules": {
             "compatibleTypes": "->Model->"
         },
-        "propertyNameAtParent": "model",
+        "propertyNameAtParent": "compile",
         "childrenNodesProperties": [
             {
                 "name": "optimizer",
@@ -517,7 +517,7 @@
         "attachingRules": {
             "compatibleTypes": "->Model->"
         },
-        "propertyNameAtParent": "model",
+        "propertyNameAtParent": "fitDataset",
         "childrenNodesProperties": [
             {
                 "name": "verbose",