diff --git a/context.go b/context.go
index 80a6bd7d..4c53e487 100644
--- a/context.go
+++ b/context.go
@@ -135,6 +135,7 @@ type SwarmNode struct {
 	ID      string
 	Name    string
 	Address Address
+	Labels  map[string]string
 }
 
 type Mount struct {
diff --git a/generator.go b/generator.go
index bf68be66..8f7afd59 100644
--- a/generator.go
+++ b/generator.go
@@ -434,6 +434,7 @@ func (g *generator) getContainers() ([]*RuntimeContainer, error) {
 			runtimeContainer.Node.Address = Address{
 				IP: container.Node.IP,
 			}
+			runtimeContainer.Node.Labels = container.Node.Labels
 		}
 
 		for _, v := range container.Mounts {