diff --git a/budget.go b/budget.go index 4bf8a22..7f995d1 100644 --- a/budget.go +++ b/budget.go @@ -19,6 +19,6 @@ type UserBudget struct { // BudgetService provides Methods for CRUD of Budgets type BudgetService interface { - Budget(id string) (Budget, error) - BudgetsForUser(id string) ([]Budget, error) + Budget(id string) (*Budget, error) + BudgetsForUser(id string) ([]*Budget, error) } diff --git a/http/data.go b/http/data.go index b866764..81bc013 100644 --- a/http/data.go +++ b/http/data.go @@ -12,8 +12,8 @@ func (d *TemplateData) GetToken() jwt.Token { return d.token } -func (d *TemplateData) GetBudgets() []budgeteer.Budget { - userID := d.token.GetID() +func (d TemplateData) GetBudgets() []*budgeteer.Budget { + userID := d.Token.GetID() budgets, err := d.budgetService.BudgetsForUser(userID) if err != nil { return nil