diff --git a/cosmos_genesis_tinker.py b/cosmos_genesis_tinker.py
index 8a253b7..e1e4311 100755
--- a/cosmos_genesis_tinker.py
+++ b/cosmos_genesis_tinker.py
@@ -562,8 +562,8 @@ def set_max_deposit_period(self, max_deposit_period: str = "1209600s"):
         self.log_step(
             "Swapping governance max deposit period to " + max_deposit_period)
 
-        deposit_params = self.app_state["gov"]["params"]
-        deposit_params["max_deposit_period"] = max_deposit_period
+        params = self.gov["params"]
+        params["max_deposit_period"] = max_deposit_period
 
         return self
 
@@ -576,8 +576,8 @@ def set_min_deposit(self, min_amount: str = "64000000", denom: str = "uatom"):
         self.log_step(
             "Swapping min governance deposit amount to " + min_amount + denom)
 
-        deposit_params = self.app_state['gov']['params']
-        min_deposit = deposit_params['min_deposit']
+        params = self.gov["params"]
+        min_deposit = params["min_deposit"]
 
         has_found_deposit_denom = False
 
@@ -602,7 +602,7 @@ def set_tally_param(self, parameter_name: str, value: str):
         self.log_step("Swapping tally parameter " +
                       parameter_name + " to " + value)
 
-        self.app_state["gov"]["tally_params"][parameter_name] = value
+        self.gov["params"][parameter_name] = value
 
         return self
 
@@ -613,7 +613,7 @@ def set_voting_period(self, voting_period: str = "1209600s"):
 
         self.log_step("Swapping governance voting period to " + voting_period)
 
-        self.app_state["gov"]["params"]["voting_period"] = voting_period
+        self.gov["params"]["voting_period"] = voting_period
 
         return self