diff --git a/entityv2/queryStruct.go b/entityv2/queryStruct.go
index 6bbaeab69b9ec4fb05af6a6ddcf89209f5893362..a0a691a6c271c4c8eec79f94d5d99138ca19d7cc 100644
--- a/entityv2/queryStruct.go
+++ b/entityv2/queryStruct.go
@@ -17,11 +17,19 @@ type IncomingQueryJSON struct {
 	// Entities        []QueryEntityStruct   `json:"entities"`
 	// Relations       []QueryRelationStruct `json:"relations"`
 	// GroupBys        []QueryGroupByStruct  `json:"groupBys"`
-	// MachineLearning []QueryMLStruct       `json:"machineLearning"`
+	MachineLearning []QueryMLStruct       `json:"machineLearning"`
 	// Modifiers []QueryModifierStruct
 	// Prefix    string
 }
 
+/*
+QueryMLStruct holds info for machinelearning
+*/
+type QueryMLStruct struct {
+	Type       string
+	Parameters []string
+}
+
 type QueryStruct struct {
 	ID        string     `json:"id"`
 	Node      NodeStruct `json:"node"`