diff --git a/api/graphql/models/gen_models.go b/api/graphql/models/gen_models.go
index 5bf8ea9ec232a96bf79bc56d67c8f7e41f5d7555..5a120f572deb7d30900bbd4c3a1db6c5541df6ae 100644
--- a/api/graphql/models/gen_models.go
+++ b/api/graphql/models/gen_models.go
@@ -8,6 +8,7 @@ import (
 	"strconv"
 
 	"github.com/MichaelMure/git-bug/bug"
+	"github.com/MichaelMure/git-bug/entity"
 	"github.com/MichaelMure/git-bug/repository"
 )
 
@@ -45,7 +46,9 @@ type EditCommentInput struct {
 	RepoRef *string `json:"repoRef"`
 	// The bug ID's prefix.
 	Prefix string `json:"prefix"`
-	// The first message of the new bug.
+	// Target
+    Target entity.Id `json:"target"`
+	// The new message to be set.
 	Message string `json:"message"`
 	// The collection of file's hash required for the first message.
 	Files []repository.Hash `json:"files"`
diff --git a/api/graphql/resolvers/mutation.go b/api/graphql/resolvers/mutation.go
index 59e93cdddf6ecde65abb1c26fc370a31331350ef..60746b8bd58d3f06c9f5796db202f18c892a9a99 100644
--- a/api/graphql/resolvers/mutation.go
+++ b/api/graphql/resolvers/mutation.go
@@ -100,7 +100,7 @@ func (r mutationResolver) EditComment(ctx context.Context, input models.EditComm
 		return nil, err
 	}
 
-	op, err := b.EditCommentRaw(author, time.Now().Unix(), input.Message, input.Files, nil)
+	op, err := b.EditCommentRaw(author, time.Now().Unix(), input.Target, input.Message, nil)
 	if err != nil {
 		return nil, err
 	}