1 files changed, 2 insertions, 8 deletions
diff --git a/middleware/kubernetes/ns_test.go b/middleware/kubernetes/ns_test.go
index c7500a3fc..1aa4f910d 100644
--- a/middleware/kubernetes/ns_test.go
+++ b/middleware/kubernetes/ns_test.go
@@ -96,19 +96,13 @@ func (APIConnTest) EndpointsList() api.EndpointsList {
 
 func (APIConnTest) GetNodeByName(name string) (api.Node, error) { return api.Node{}, nil }
 
-type interfaceAddrsTest struct{}
-
-func (i interfaceAddrsTest) interfaceAddrs() ([]net.Addr, error) {
-	_, ipnet, _ := net.ParseCIDR("172.0.40.10/32")
-	return []net.Addr{ipnet}, nil
-}
-
 func TestDoCoreDNSRecord(t *testing.T) {
 
 	corednsRecord = dns.A{}
 	k := Kubernetes{Zones: []string{"inter.webs.test"}}
 
-	k.interfaceAddrs = &interfaceAddrsTest{}
+	k.interfaceAddrsFunc = func() net.IP { return net.ParseIP("172.0.40.10") }
+
 	k.APIConn = &APIConnTest{}
 
 	cdr := k.coreDNSRecord()