fix non deterministic tests

cosmos · Mar 8, 2024 · 8710435 · 8710435
1 parent e195287
commit 8710435
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/migrate_test.go b/migrate_test.go
@@ -124,8 +124,10 @@ func TestLegacyReferenceNode(t *testing.T) {
 	_, err = newTree.LoadVersion(version - 1)
 	require.NoError(t, err)
 	// Check if the reference node is refactored
+	legacyLatestVersion, err := newTree.ndb.getLegacyLatestVersion()
+	require.NoError(t, err)
 	require.Equal(t, newTree.root.nodeKey.nonce, uint32(0))
-	require.Equal(t, newTree.root.nodeKey.version, int64(legacyVersion))
+	require.Equal(t, newTree.root.nodeKey.version, int64(legacyLatestVersion))
 }
 
 func TestDeleteVersions(t *testing.T) {