diff --git a/contracts/GovernanceAggregator.sol b/contracts/GovernanceAggregator.sol index c8af3f7..221f29c 100644 --- a/contracts/GovernanceAggregator.sol +++ b/contracts/GovernanceAggregator.sol @@ -60,6 +60,7 @@ contract GovernanceAggregator { returns ( uint256 balance, uint256 latestProposalId, + uint256 latestProposalIdState uint256 timelock, address delegatee ) @@ -67,6 +68,9 @@ contract GovernanceAggregator { // Core core = Core(address(governance)); balance = governance.lockedBalance(account); latestProposalId = governance.latestProposalIds(account); + if(latestProposalId != 0) { + latestProposalIdState = governance.state(latestProposalId) + } timelock = governance.canWithdrawAfter(account); delegatee = governance.delegatedTo(account); }