diff --git a/pkg/provider/docker/pswarm_test.go b/pkg/provider/docker/pswarm_test.go
index a4fd9692e..f762f884d 100644
--- a/pkg/provider/docker/pswarm_test.go
+++ b/pkg/provider/docker/pswarm_test.go
@@ -358,6 +358,7 @@ func TestSwarmProvider_parseService_task(t *testing.T) {
 			},
 			nodes: []swarmtypes.Node{
 				{
+					ID: "id1",
 					Status: swarmtypes.NodeStatus{
 						Addr: "10.11.12.13",
 					},
@@ -387,7 +388,10 @@ func TestSwarmProvider_parseService_task(t *testing.T) {
 			dData, err := p.parseService(t.Context(), test.service, test.networks)
 			require.NoError(t, err)
 
-			dockerClient := &fakeServicesClient{tasks: test.tasks}
+			dockerClient := &fakeServicesClient{
+				tasks: test.tasks,
+				nodes: test.nodes,
+			}
 
 			for _, task := range test.tasks {
 				taskDockerData, err := parseTasks(t.Context(), dockerClient, task, dData, test.networks, test.isGlobalSVC)
@@ -395,6 +399,7 @@ func TestSwarmProvider_parseService_task(t *testing.T) {
 
 				expected := test.expected[task.ID]
 				assert.Equal(t, expected.Name, taskDockerData.Name)
+				assert.Equal(t, expected.NodeIP, taskDockerData.NodeIP)
 			}
 		})
 	}