diff --git a/logic/models/edit.go b/logic/models/edit.go
index 256d270..63cb37f 100644
--- a/logic/models/edit.go
+++ b/logic/models/edit.go
@@ -164,12 +164,12 @@ func handleEdit(handle *Handle) {
 							return c.Error500(err)
 						}
 
-                        c.Logger.Info("res", "id", lastLayer.Id, "start", lastLayer.Range_start, "end", lastLayer.Range_end)
+                        c.Logger.Info("res", "id", lastLayer.Id, "start", lastLayer.Range_start, "end", lastLayer.Range_end, "shape", fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start + 1))
 
 						layers = append(layers, layerdef{
 							id:         lastLayer.Id,
 							LayerType: LAYER_DENSE,
-							Shape:     fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start),
+							Shape:     fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start + 1),
 						})
 
 						/*
diff --git a/logic/models/train/train.go b/logic/models/train/train.go
index ef83c76..ecd9cb4 100644
--- a/logic/models/train/train.go
+++ b/logic/models/train/train.go
@@ -283,7 +283,7 @@ func trainDefinitionExp(c *Context, model *BaseModel, definition_id string, load
 
 	got = append(got, layerrow{
 		LayerType: LAYER_DENSE,
-		Shape:     fmt.Sprintf("%d", exp.end-exp.start),
+		Shape:     fmt.Sprintf("%d", exp.end-exp.start + 1),
 		ExpType:   2,
 		LayerNum:  i,
 	})