diff --git a/gossipsub_connmgr_test.go b/gossipsub_connmgr_test.go
index 67fe619..bd12f41 100644
--- a/gossipsub_connmgr_test.go
+++ b/gossipsub_connmgr_test.go
@@ -2,11 +2,12 @@ package pubsub
 
 import (
 	"context"
+	"testing"
+	"time"
+
 	"github.com/benbjohnson/clock"
 	"github.com/libp2p/go-libp2p-core/host"
 	swarmt "github.com/libp2p/go-libp2p-swarm/testing"
-	"testing"
-	"time"
 
 	bhost "github.com/libp2p/go-libp2p-blankhost"
 	connmgr "github.com/libp2p/go-libp2p-connmgr"
diff --git a/tag_tracer_test.go b/tag_tracer_test.go
index 23c4840..b664945 100644
--- a/tag_tracer_test.go
+++ b/tag_tracer_test.go
@@ -2,13 +2,15 @@ package pubsub
 
 import (
 	"fmt"
+	"testing"
+	"time"
+
 	"github.com/benbjohnson/clock"
 	connmgr "github.com/libp2p/go-libp2p-connmgr"
 	connmgri "github.com/libp2p/go-libp2p-core/connmgr"
 	"github.com/libp2p/go-libp2p-core/peer"
+
 	pb "github.com/libp2p/go-libp2p-pubsub/pb"
-	"testing"
-	"time"
 )
 
 func TestTagTracerMeshTags(t *testing.T) {