diff --git a/gnovm/cmd/gno/test.go b/gnovm/cmd/gno/test.go
index 8d66236ab2f..e6c77817f6a 100644
--- 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 {