diff --git a/packages/taiko-client/driver/driver.go b/packages/taiko-client/driver/driver.go
index 49e6b851d1c..20e81f44c6b 100644
--- a/packages/taiko-client/driver/driver.go
+++ b/packages/taiko-client/driver/driver.go
@@ -127,6 +127,8 @@ func (d *Driver) InitFromConfig(ctx context.Context, cfg *Config) (err error) {
 
 	if cfg.P2PConfigs != nil {
 		log.Info("enabling p2p network")
+		d.p2pSetup = cfg.P2PConfigs
+
 		if d.p2pNode, err = p2p.NewNodeP2P(
 			d.ctx,
 			&rollup.Config{L1ChainID: d.rpc.L1.ChainID, L2ChainID: d.rpc.L2.ChainID},