diff --git a/ecs/pkg/amazon/sdk/api.go b/ecs/pkg/amazon/sdk/api.go
index 1e3e61ad3..feaa1eb77 100644
--- a/ecs/pkg/amazon/sdk/api.go
+++ b/ecs/pkg/amazon/sdk/api.go
@@ -8,7 +8,7 @@ import (
 	"github.com/docker/ecs-plugin/pkg/compose"
 )
 
-//go:generate mockgen -destination=./api_mock.go -self_package "github.com/docker/ecs-plugin/pkg/amazon" -package=amazon . API
+//go:generate mockgen -destination=./api_mock.go -self_package "github.com/docker/ecs-plugin/pkg/amazon/sdk" -package=sdk . API
 
 type API interface {
 	downAPI