tasks: Fix renounce role to run before registration

livepeer · Oct 11, 2023 · 021ae03 · 021ae03
1 parent f21b583
commit 021ae03
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/tasks/treasury-renounce-admin-role.ts b/tasks/treasury-renounce-admin-role.ts
@@ -1,6 +1,5 @@
 import {task} from "hardhat/config"
-import {Controller, Treasury} from "../typechain"
-import {contractId} from "../utils/helpers"
+import {Treasury} from "../typechain"
 
 task(
     "treasury-renounce-admin-role",
@@ -10,15 +9,12 @@ task(
 
     const {deployer} = await hre.getNamedAccounts()
 
-    const controller = await deployments.get("Controller")
-    const Controller: Controller = await hre.ethers.getContractAt(
-        "Controller",
-        controller.address
+    const treasury = await deployments.get("Treasury")
+    const Treasury: Treasury = await ethers.getContractAt(
+        "Treasury",
+        treasury.address
     )
 
-    const address = await Controller.getContract(contractId("Treasury"))
-    const Treasury: Treasury = await ethers.getContractAt("Treasury", address)
-
     const adminRole = await Treasury.TIMELOCK_ADMIN_ROLE()
     let hasAdminRole = await Treasury.hasRole(adminRole, deployer)
     if (!hasAdminRole) {