diff --git a/.tool-versions b/.tool-versions
index 4267afd..35b9c2b 100644
--- a/.tool-versions
+++ b/.tool-versions
@@ -1,2 +1,2 @@
-scarb 2.4.4
+scarb 2.5.0
 starknet-foundry 0.16.0
diff --git a/Scarb.toml b/Scarb.toml
index a4a5e93..dd0a9d9 100644
--- a/Scarb.toml
+++ b/Scarb.toml
@@ -5,7 +5,7 @@ edition = "2023_10"
 
 [dependencies]
 snforge_std = { git = "https://github.com/foundry-rs/starknet-foundry", tag = "v0.16.0" }
-starknet = "2.4.4"
+starknet = "2.5.0"
 
 [[target.starknet-contract]]
 casm = true
diff --git a/tests/test_contract.cairo b/tests/test_contract.cairo
index e13999a..df35f17 100644
--- a/tests/test_contract.cairo
+++ b/tests/test_contract.cairo
@@ -33,9 +33,11 @@ fn test_cannot_increase_balance_with_zero_value() {
 
     let safe_dispatcher = IHelloStarknetSafeDispatcher { contract_address };
 
+    #[feature("safe_dispatcher")]
     let balance_before = safe_dispatcher.get_balance().unwrap();
     assert(balance_before == 0, 'Invalid balance');
 
+    #[feature("safe_dispatcher")]
     match safe_dispatcher.increase_balance(0) {
         Result::Ok(_) => panic_with_felt252('Should have panicked'),
         Result::Err(panic_data) => {