diff --git a/pkg/training/repository.go b/pkg/training/repository.go
index 986a72e..d3e7974 100644
--- a/pkg/training/repository.go
+++ b/pkg/training/repository.go
@@ -4,7 +4,7 @@ type TrainingRepository interface {
 	Create(training *Training) error
 	FindByID(id TrainingID) (*Training, error)
 	FindAll() ([]Training, error)
-	Update(id TrainingID, training *Training) error
+	Update(training *Training) error
 	Delete(id TrainingID) error
 }
 
@@ -68,8 +68,8 @@ func (r *InMemoryTrainingRepository) FindAll() ([]Training, error) {
 	return trainings, nil
 }
 
-func (r *InMemoryTrainingRepository) Update(id TrainingID, training *Training) error {
-	r.trainings[id] = *training
+func (r *InMemoryTrainingRepository) Update(training *Training) error {
+	r.trainings[training.ID] = *training
 	return nil
 }