diff --git a/pkg/transport/transport.go b/pkg/transport/transport.go
index de5eda7..e983377 100644
--- a/pkg/transport/transport.go
+++ b/pkg/transport/transport.go
@@ -323,6 +323,12 @@ func GetTransportInstance() *Transport {
 	return transportInstance
 }
 
+func SetTransportInstance(t *Transport) {
+	transportMutex.Lock()
+	defer transportMutex.Unlock()
+	transportInstance = t
+}
+
 func (t *Transport) RegisterInterface(name string, iface common.NetworkInterface) error {
 	t.mutex.Lock()
 	defer t.mutex.Unlock()