Fix tests?

tests/flows/teleporter_registry.go

@@ -4,6 +4,7 @@ import (
 	"context"
 
 	"github.com/ava-labs/subnet-evm/accounts/abi/bind"
+	teleportermessenger "github.com/ava-labs/teleporter/abi-bindings/go/teleporter/TeleporterMessenger"
 	"github.com/ava-labs/teleporter/tests/interfaces"
 	"github.com/ava-labs/teleporter/tests/utils"
 	. "github.com/onsi/gomega"
@@ -125,6 +126,9 @@ func TeleporterRegistry(network interfaces.LocalNetwork) {
 	Expect(minTeleporterVersionUpdatedEvent.OldMinTeleporterVersion.Cmp(minTeleporterVersion)).Should(Equal(0))
 	Expect(minTeleporterVersionUpdatedEvent.NewMinTeleporterVersion.Cmp(latestVersionB)).Should(Equal(0))
 
+	cChainInfo = network.GetPrimaryNetworkInfo()
+	subnetAInfo, subnetBInfo = utils.GetTwoSubnets(network)
+
 	// Send a message using old Teleporter version to test messenger with updated minimum Teleporter version.
 	// Message should fail since we updated minimum Teleporter version.
 	utils.SendExampleCrossChainMessageAndVerify(
@@ -142,8 +146,16 @@ func TeleporterRegistry(network interfaces.LocalNetwork) {
 
 	// Update the subnets to use new Teleporter messengers
 	network.SetTeleporterContractAddress(newTeleporterAddress)
-	cChainInfo = network.GetPrimaryNetworkInfo()
-	subnetAInfo, subnetBInfo = utils.GetTwoSubnets(network)
+	teleporterMessengerA, err := teleportermessenger.NewTeleporterMessenger(
+		newTeleporterAddress, subnetAInfo.RPCClient,
+	)
+	Expect(err).Should(BeNil())
+	subnetAInfo.TeleporterMessenger = teleporterMessengerA
+	teleporterMessengerC, err := teleportermessenger.NewTeleporterMessenger(
+		newTeleporterAddress, cChainInfo.RPCClient,
+	)
+	Expect(err).Should(BeNil())
+	cChainInfo.TeleporterMessenger = teleporterMessengerC
 
 	utils.SendExampleCrossChainMessageAndVerify(
 		ctx,