From e422b395b6187569e76e3b2d0736b58f2ba40880 Mon Sep 17 00:00:00 2001
From: DaveVodrazka <dave.vodrazka@gmail.com>
Date: Tue, 12 Dec 2023 10:25:16 +0100
Subject: [PATCH] fix: use get_block_timestamp

---
 src/proposals.cairo | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/proposals.cairo b/src/proposals.cairo
index f6ee6b59..ba0ac4e9 100644
--- a/src/proposals.cairo
+++ b/src/proposals.cairo
@@ -14,6 +14,7 @@ mod Proposals {
 
     use starknet::contract_address::ContractAddressZeroable;
     use starknet::get_block_info;
+    use starknet::get_block_timestamp;
     use starknet::get_caller_address;
     use starknet::BlockInfo;
     use starknet::ContractAddress;
@@ -65,7 +66,7 @@ mod Proposals {
         let end_timestamp: u64 = state.proposal_vote_end_timestamp.read(prop_id);
         assert(end_timestamp != 0, 'prop_id not found');
 
-        let current_timestamp: u64 = get_block_info().unbox().block_timestamp;
+        let current_timestamp: u64 = get_block_timestamp();
 
         assert(end_timestamp > current_timestamp, 'voting concluded');
     }
@@ -120,7 +121,7 @@ mod Proposals {
         let prop_details = PropDetails { payload: payload, to_upgrade: to_upgrade };
         state.proposal_details.write(prop_id, prop_details);
 
-        let current_timestamp: u64 = get_block_info().unbox().block_timestamp;
+        let current_timestamp: u64 = get_block_timestamp();
         let end_timestamp: u64 = current_timestamp + constants::PROPOSAL_VOTING_SECONDS;
         state.proposal_vote_end_timestamp.write(prop_id, end_timestamp);
 
@@ -316,7 +317,7 @@ mod Proposals {
         let state = Governance::unsafe_new_contract_state();
 
         let end_timestamp: u64 = state.proposal_vote_end_timestamp.read(prop_id);
-        let current_timestamp: u64 = get_block_info().unbox().block_timestamp;
+        let current_timestamp: u64 = get_block_timestamp();
 
         if current_timestamp <= end_timestamp {
             return check_proposal_passed_express(prop_id).into();