Update module.go

cosmos · Sep 13, 2023 · 0aa8200 · 0aa8200
1 parent a21eefc
commit 0aa8200
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/x/ccv/consumer/module.go b/x/ccv/consumer/module.go
@@ -107,9 +107,11 @@ func (AppModule) RegisterInvariants(ir sdk.InvariantRegistry) {
 // RegisterServices registers module services.
 func (am AppModule) RegisterServices(cfg module.Configurator) {
 	consumertypes.RegisterQueryServer(cfg.QueryServer(), am.keeper)
-	cfg.RegisterMigration(consumertypes.ModuleName,
-		1, // from version 1
-		am.keeper.MigrateConsumerPacketData)
+
+	m := keeper.NewMigrator(am.keeper, am.paramSpace)
+	if err := cfg.RegisterMigration(consumertypes.ModuleName, 1, m.Migrate1to2); err != nil {
+		panic(fmt.Sprintf("failed to register migrator for %s: %s", consumertypes.ModuleName, err))
+	}
 }
 
 // InitGenesis performs genesis initialization for the consumer module. It returns