diff --git a/integration_tests/test_upgrade.py b/integration_tests/test_upgrade.py
index 32e309566..2937373f0 100644
--- a/integration_tests/test_upgrade.py
+++ b/integration_tests/test_upgrade.py
@@ -427,7 +427,7 @@ def assert_commission(adr, expected):
     for i in range(2):
         cluster.migrate_keystore(i=i)
     upgrade(cluster, "v4.3.0", target_height)
-
+    cli = cluster.cosmos_cli()
     target_height = cluster.block_height() + 15
     gov_param = cli.query_params("gov")
     upgrade(cluster, "v5.0", target_height, broadcast_mode="sync")