diff --git a/deploy/60_validate_upgrade_token_staking.ts b/deploy/60_validate_upgrade_token_staking.ts
index 6ed2a07..6961844 100644
--- a/deploy/60_validate_upgrade_token_staking.ts
+++ b/deploy/60_validate_upgrade_token_staking.ts
@@ -8,20 +8,12 @@ const func: DeployFunction = async function (hre: HardhatRuntimeEnvironment) {
   const { log } = deployments
 
   const T = await deployments.get("T")
-  const KeepTokenStaking = await deployments.get("KeepTokenStaking")
-  const NuCypherStakingEscrow = await deployments.get("NuCypherStakingEscrow")
-  const VendingMachineKeep = await deployments.get("VendingMachineKeep")
   const VendingMachineNuCypher = await deployments.get("VendingMachineNuCypher")
-  const KeepStake = await deployments.get("KeepStake")
   const TokenStakingDeployment = await deployments.get("TokenStaking")
 
   const tokenStakingConstructorArgs = [
     T.address,
-    KeepTokenStaking.address,
-    NuCypherStakingEscrow.address,
-    VendingMachineKeep.address,
     VendingMachineNuCypher.address,
-    KeepStake.address,
   ]
 
   if (hre.network.name == "mainnet") {
diff --git a/deploy/61_prepare_upgrade_token_staking.ts b/deploy/61_prepare_upgrade_token_staking.ts
index 643550d..768190a 100644
--- a/deploy/61_prepare_upgrade_token_staking.ts
+++ b/deploy/61_prepare_upgrade_token_staking.ts
@@ -8,20 +8,12 @@ const func: DeployFunction = async function (hre: HardhatRuntimeEnvironment) {
   const { log } = deployments
 
   const T = await deployments.get("T")
-  const KeepTokenStaking = await deployments.get("KeepTokenStaking")
-  const NuCypherStakingEscrow = await deployments.get("NuCypherStakingEscrow")
-  const VendingMachineKeep = await deployments.get("VendingMachineKeep")
   const VendingMachineNuCypher = await deployments.get("VendingMachineNuCypher")
-  const KeepStake = await deployments.get("KeepStake")
   const TokenStakingProxy = await deployments.get("TokenStaking")
 
   const tokenStakingConstructorArgs = [
     T.address,
-    KeepTokenStaking.address,
-    NuCypherStakingEscrow.address,
-    VendingMachineKeep.address,
     VendingMachineNuCypher.address,
-    KeepStake.address,
   ]
 
   if (hre.network.name == "mainnet") {
@@ -65,3 +57,4 @@ const func: DeployFunction = async function (hre: HardhatRuntimeEnvironment) {
 export default func
 
 func.tags = ["PrepareUpgradeTokenStaking"]
+func.dependencies = ["ValidateUpgradeTokenStaking"]