From 393a0aa086480c2ddc668ff8f803b6a7ed765511 Mon Sep 17 00:00:00 2001
From: yihuang <huang@crypto.com>
Date: Fri, 18 Oct 2024 17:10:02 +0800
Subject: [PATCH] Update testground/benchmark/benchmark/cosmostx.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: yihuang <huang@crypto.com>
---
 testground/benchmark/benchmark/cosmostx.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/testground/benchmark/benchmark/cosmostx.py b/testground/benchmark/benchmark/cosmostx.py
index ba7eb07d25..587aa84742 100644
--- a/testground/benchmark/benchmark/cosmostx.py
+++ b/testground/benchmark/benchmark/cosmostx.py
@@ -38,8 +38,8 @@ class ModeInfoMulti(ProtoEntity):
 
 
 class ModeInfo(ProtoEntity):
-    single = Field("ModeInfo.Single", 1)
-    multi = Field("ModeInfo.Multi", 2)
+    single = Field(ModeInfoSingle, 1)
+    multi = Field(ModeInfoMulti, 2)
 
 
 class SignerInfo(ProtoEntity):