diff --git a/swarm.go b/swarm.go
index 4eebf626bea5f62145938900b1cca00036568fba..19ccf13eef72908fc4407e4dc01059ae829f5a61 100644
--- a/swarm.go
+++ b/swarm.go
@@ -15,7 +15,6 @@ import (
 	mconn "github.com/libp2p/go-libp2p/p2p/metrics/conn"
 	inet "github.com/libp2p/go-libp2p/p2p/net"
 	conn "github.com/libp2p/go-libp2p/p2p/net/conn"
-	filter "github.com/libp2p/go-libp2p/p2p/net/filter"
 	addrutil "github.com/libp2p/go-libp2p/p2p/net/swarm/addr"
 
 	ci "github.com/ipfs/go-libp2p-crypto"
@@ -28,6 +27,7 @@ import (
 	"github.com/jbenet/goprocess"
 	goprocessctx "github.com/jbenet/goprocess/context"
 	transport "github.com/libp2p/go-libp2p-transport"
+	filter "github.com/libp2p/go-maddr-filter"
 	tcpt "github.com/libp2p/go-tcp-transport"
 	psmss "github.com/whyrusleeping/go-smux-multistream"
 	spdy "github.com/whyrusleeping/go-smux-spdystream"
diff --git a/swarm_stream.go b/swarm_stream.go
index dc365e8c092c79d439daf532980412d88e07008d..51c846001a5df43c85e7ff7e445459ead056d85b 100644
--- a/swarm_stream.go
+++ b/swarm_stream.go
@@ -1,8 +1,8 @@
 package swarm
 
 import (
+	protocol "github.com/libp2p/go-libp2p-protocol"
 	inet "github.com/libp2p/go-libp2p/p2p/net"
-	protocol "github.com/libp2p/go-libp2p/p2p/protocol"
 
 	ps "github.com/jbenet/go-peerstream"
 )