1 files changed, 9 insertions, 8 deletions
diff --git a/graphql/resolvers/query.go b/graphql/resolvers/query.go
index cceca334..bca4806f 100644
--- a/graphql/resolvers/query.go
+++ b/graphql/resolvers/query.go
@@ -3,34 +3,35 @@ package resolvers
 import (
 	"context"
 	"github.com/MichaelMure/git-bug/cache"
+	"github.com/MichaelMure/git-bug/graphql/models"
 )
 
 type rootQueryResolver struct {
 	cache cache.Cacher
 }
 
-func (r rootQueryResolver) DefaultRepository(ctx context.Context) (*repoResolver, error) {
+func (r rootQueryResolver) DefaultRepository(ctx context.Context) (*models.Repository, error) {
 	repo, err := r.cache.DefaultRepo()
 
 	if err != nil {
 		return nil, err
 	}
 
-	return &repoResolver{
-		cache: r.cache,
-		repo:  repo,
+	return &models.Repository{
+		Cache: r.cache,
+		Repo:  repo,
 	}, nil
 }
 
-func (r rootQueryResolver) Repository(ctx context.Context, id string) (*repoResolver, error) {
+func (r rootQueryResolver) Repository(ctx context.Context, id string) (*models.Repository, error) {
 	repo, err := r.cache.ResolveRepo(id)
 
 	if err != nil {
 		return nil, err
 	}
 
-	return &repoResolver{
-		cache: r.cache,
-		repo:  repo,
+	return &models.Repository{
+		Cache: r.cache,
+		Repo:  repo,
 	}, nil
 }