Use function_selector to get the appropriate value

OpenZeppelin · Sep 19, 2024 · 025564a · 025564a
1 parent 9e449f4
commit 025564a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/contracts/src/token/erc20/utils/safe_erc20.rs b/contracts/src/token/erc20/utils/safe_erc20.rs
@@ -2,16 +2,16 @@
 
 use alloc::vec::Vec;
 
-use alloy_primitives::{keccak256, Address, U256};
+use alloy_primitives::{Address, U256};
 use alloy_sol_types::{
     sol,
     sol_data::{Address as SOLAddress, Uint},
     SolType,
 };
 use stylus_proc::SolidityError;
 use stylus_sdk::{
-    call::RawCall, contract::address, storage::TopLevelStorage,
-    types::AddressVM,
+    call::RawCall, contract::address, function_selector,
+    storage::TopLevelStorage, types::AddressVM,
 };
 
 use crate::token::{erc20, erc20::Erc20};
@@ -67,7 +67,8 @@ impl SafeErc20 for Erc20 {
         type TransferType = (SOLAddress, Uint<256>);
         let tx_data = (to, value);
         let data = TransferType::abi_encode_params(&tx_data);
-        let hashed_function_selector = keccak256("transfer".as_bytes());
+        let hashed_function_selector =
+            function_selector!("transfer", Address, U256);
         // Combine function selector and input data (use abi_packed way)
         let calldata = [&hashed_function_selector[..4], &data].concat();