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):