diff --git a/packages/gossip/manager_test.go b/packages/gossip/manager_test.go
index 7674e2b281eafcf603ff15df1ee1020a9c30cce7..ea9ca56b98a13460ea8e3a3221c26f89e7498e82 100644
--- a/packages/gossip/manager_test.go
+++ b/packages/gossip/manager_test.go
@@ -6,11 +6,11 @@ import (
 	"testing"
 	"time"
 
-	pb "github.com/iotaledger/goshimmer/packages/gossip/proto"
-	"github.com/iotaledger/goshimmer/packages/gossip/transport"
 	"github.com/golang/protobuf/proto"
 	"github.com/iotaledger/autopeering-sim/peer"
 	"github.com/iotaledger/autopeering-sim/peer/service"
+	pb "github.com/iotaledger/goshimmer/packages/gossip/proto"
+	"github.com/iotaledger/goshimmer/packages/gossip/transport"
 	"github.com/iotaledger/hive.go/events"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
@@ -118,7 +118,7 @@ func TestClosedConnection(t *testing.T) {
 	eventMock.On("neighborDroppedEvent", &NeighborDroppedEvent{Peer: peerB}).Once()
 
 	// A drops B
-	err := mgrA.NeighborDropped(peerB.ID())
+	err := mgrA.DropNeighbor(peerB.ID())
 	require.NoError(t, err)
 	time.Sleep(graceTime)
 
diff --git a/packages/gossip/transport/handshake.go b/packages/gossip/transport/handshake.go
index 661443e8f6b660b1ba7f2bfc7fdc4d51d756c436..e5e928c4e2ca079fbc6dbc61cd82e43f8c27bce1 100644
--- a/packages/gossip/transport/handshake.go
+++ b/packages/gossip/transport/handshake.go
@@ -4,9 +4,9 @@ import (
 	"bytes"
 	"time"
 
-	pb "github.com/iotaledger/goshimmer/packages/gossip/transport/proto"
 	"github.com/golang/protobuf/proto"
 	"github.com/iotaledger/autopeering-sim/server"
+	pb "github.com/iotaledger/goshimmer/packages/gossip/transport/proto"
 )
 
 const (