better use the machine API

gnolang · Jun 21, 2023 · ddbab8a · ddbab8a
1 parent 18eb207
commit ddbab8a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gnovm/cmd/gno/test.go b/gnovm/cmd/gno/test.go
@@ -383,8 +383,8 @@ func runTestFiles(
 	m.RunFiles(n)
 	// XXX Affect an ID to testmain so it doesn't trigger a panic when
 	// realm.FinalizeRealmTransaction() is invoked
-	oo := m.Package.FBlocks[len(m.Package.FBlocks)-1]
-	m.Realm.AssignNewObjectID(oo.(*gno.Block))
+	oo := m.Package.GetFileBlock(m.Store, n.Name)
+	m.Realm.AssignNewObjectID(oo)
 
 	for _, test := range testFuncs.Tests {
 		if verbose {