diff --git a/contracts/test/TACoApplicationTestSet.sol b/contracts/test/TACoApplicationTestSet.sol
index 5e980bc4..3fbc6ac0 100644
--- a/contracts/test/TACoApplicationTestSet.sol
+++ b/contracts/test/TACoApplicationTestSet.sol
@@ -4,7 +4,7 @@ pragma solidity ^0.8.0;
 
 
 import "../contracts/TACoApplication.sol";
-import "../threshold/IApplication.sol";
+import "@threshold/contracts/staking/IApplication.sol";
 import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
 
 
diff --git a/tests/application/test_reward.py b/tests/application/test_reward.py
index 8c9747e8..df2ad775 100644
--- a/tests/application/test_reward.py
+++ b/tests/application/test_reward.py
@@ -113,7 +113,7 @@ def test_push_reward(accounts, token, threshold_staking, taco_application, chain
     assert taco_application.lastTimeRewardApplicable() == timestamp
     assert taco_application.rewardPerTokenStored() == 0
     assert taco_application.rewardPerToken() == 0
-    assert taco_application.availableRewardsed(staking_provider_1) == 0
+    assert taco_application.availableRewards(staking_provider_1) == 0
 
     events = taco_application.RewardAdded.from_receipt(tx)
     assert len(events) == 1
@@ -426,7 +426,7 @@ def test_withdraw(accounts, token, threshold_staking, taco_application, chain):
     # Withdraw
     chain.pending_timestamp += reward_duration // 2
     assert taco_application.availableRewards(staking_provider) == new_earned
-    tx = taco_application.withdraw(staking_provider, sender=beneficiary)
+    tx = taco_application.withdrawRewards(staking_provider, sender=beneficiary)
     new_reward_per_token = taco_application.rewardPerToken()
     assert taco_application.rewardPerTokenStored() == new_reward_per_token
     assert taco_application.stakingProviderInfo(staking_provider)[REWARDS_SLOT] == 0