diff --git a/src/staking/RewardsDistribution.sol b/src/staking/KtonDAOVault.sol
similarity index 97%
rename from src/staking/RewardsDistribution.sol
rename to src/staking/KtonDAOVault.sol
index e5fb8c0..54f6bcb 100644
--- a/src/staking/RewardsDistribution.sol
+++ b/src/staking/KtonDAOVault.sol
@@ -7,7 +7,7 @@ import "./interfaces/IOldStakingRewards.sol";
 import "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol";
 import "@openzeppelin/contracts-upgradeable/access/Ownable2StepUpgradeable.sol";
 
-contract RewardsDistribution is Initializable, Ownable2StepUpgradeable {
+contract KtonDAOVault is Initializable, Ownable2StepUpgradeable {
     // "modlda/trsry" in bytes.
     address public constant NOTIFIER = 0x6D6f646c64612f74727372790000000000000000;
 
diff --git a/src/staking/interfaces/IStakingRewards.sol b/src/staking/interfaces/IStakingRewards.sol
index 5444c76..0f7bbd7 100644
--- a/src/staking/interfaces/IStakingRewards.sol
+++ b/src/staking/interfaces/IStakingRewards.sol
@@ -19,7 +19,7 @@ interface IStakingRewards {
 
     // Mutative
 
-    function stake(uint256 amount) external;
+    function stake(uint256 amount, address delegatee) external;
 
     function withdraw(uint256 amount) external;