Allow same VM to be assigned to the global vm. (#4236)

cosmos · Aug 3, 2023 · 2e17bb2 · 2e17bb2
1 parent c4f8da4
commit 2e17bb2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/modules/light-clients/08-wasm/keeper/keeper.go b/modules/light-clients/08-wasm/keeper/keeper.go
@@ -5,6 +5,7 @@ import (
 	"crypto/sha256"
 	"encoding/hex"
 	"fmt"
+	"reflect"
 
 	wasmvm "github.com/CosmWasm/wasmvm"
 
@@ -37,8 +38,8 @@ func NewKeeperWithVM(
 	authority string,
 	vm *wasmvm.VM,
 ) Keeper {
-	if types.WasmVM != nil {
-		panic("global Wasm VM instance should not be already set before calling this function")
+	if types.WasmVM != nil && !reflect.DeepEqual(types.WasmVM, vm) {
+		panic("global Wasm VM instance should not be set to a different instance")
 	}
 
 	types.WasmVM = vm