diff --git a/internal/dockeragent/agent.go b/internal/dockeragent/agent.go
index 9135db7..13b2823 100644
--- a/internal/dockeragent/agent.go
+++ b/internal/dockeragent/agent.go
@@ -22,7 +22,6 @@ import (
 	"syscall"
 	"time"
 
-	"github.com/docker/docker/api/types"
 	containertypes "github.com/docker/docker/api/types/container"
 	"github.com/docker/docker/api/types/filters"
 	systemtypes "github.com/docker/docker/api/types/system"
@@ -766,7 +765,7 @@ func (a *Agent) pruneStaleCPUSamples(active map[string]struct{}) {
 	}
 }
 
-func (a *Agent) collectContainer(ctx context.Context, summary types.Container) (agentsdocker.Container, error) {
+func (a *Agent) collectContainer(ctx context.Context, summary containertypes.Summary) (agentsdocker.Container, error) {
 	const perContainerTimeout = 15 * time.Second
 
 	containerCtx, cancel := context.WithTimeout(ctx, perContainerTimeout)
diff --git a/internal/dockeragent/swarm.go b/internal/dockeragent/swarm.go
index a7ac54a..8bcc95f 100644
--- a/internal/dockeragent/swarm.go
+++ b/internal/dockeragent/swarm.go
@@ -7,7 +7,6 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/filters"
 	swarmtypes "github.com/docker/docker/api/types/swarm"
 	systemtypes "github.com/docker/docker/api/types/system"
@@ -159,7 +158,7 @@ func (a *Agent) collectSwarmData(ctx context.Context, info systemtypes.Info, con
 }
 
 func (a *Agent) collectSwarmDataFromManager(ctx context.Context, info systemtypes.Info, scope string, containers map[string]agentsdocker.Container, includeServices, includeTasks bool) ([]agentsdocker.Service, []agentsdocker.Task, error) {
-	serviceList, err := a.docker.ServiceList(ctx, types.ServiceListOptions{Status: true})
+	serviceList, err := a.docker.ServiceList(ctx, swarmtypes.ServiceListOptions{Status: true})
 	if err != nil {
 		return nil, nil, err
 	}
@@ -184,7 +183,7 @@ func (a *Agent) collectSwarmDataFromManager(ctx context.Context, info systemtype
 			taskFilters.Add("node", info.Swarm.NodeID)
 		}
 
-		taskList, err := a.docker.TaskList(ctx, types.TaskListOptions{Filters: taskFilters})
+		taskList, err := a.docker.TaskList(ctx, swarmtypes.TaskListOptions{Filters: taskFilters})
 		if err != nil {
 			return services, nil, err
 		}