diff --git a/contracts/connectors/1inch.sol b/contracts/connectors/1inch.sol
index de7e1f8..18a1af4 100644
--- a/contracts/connectors/1inch.sol
+++ b/contracts/connectors/1inch.sol
@@ -74,7 +74,7 @@ contract OneHelpers is Stores, DSMath {
     /**
      * @dev Return 1proto Address
      */
-    function getOneProtoAddress() internal view returns (address payable) {
+    function getOneProtoAddress() internal virtual view returns (address payable) {
         return payable(OneProtoMappingInterface(getOneProtoMappingAddress()).oneProtoAddress());
     }
 
@@ -642,4 +642,4 @@ contract OneInch is OneProto {
 
 contract ConnectOne is OneInch {
     string public name = "1inch-1proto-v1";
-}
\ No newline at end of file
+}
diff --git a/contracts/mapping/staking.sol b/contracts/mapping/staking.sol
index f2ee950..9d9408b 100644
--- a/contracts/mapping/staking.sol
+++ b/contracts/mapping/staking.sol
@@ -83,6 +83,7 @@ contract Helpers is BytesHelper {
   ) public isChief {
     require(stakingAddress != address(0), "stakingAddress-not-vaild");
     require(stakingToken != address(0), "stakingToken-not-vaild");
+    require(rewardToken != address(0), "rewardToken-not-vaild");
     require(bytes(stakingName).length <= 32, "Length-exceeds");
     bytes32 stakeType = stringToBytes32(stakingName);
     require(stakingMapping[stakeType].stakingPool == address(0), "StakingPool-already-added");
diff --git a/contracts/tests/mockOneProto.sol b/contracts/tests/mockOneProto.sol
index 6311279..a645b52 100644
--- a/contracts/tests/mockOneProto.sol
+++ b/contracts/tests/mockOneProto.sol
@@ -1,7 +1,7 @@
 pragma solidity ^0.6.0;
 pragma experimental ABIEncoderV2;
 
-import { ConnectOne } from "../connectors/1proto.sol";
+import { ConnectOne } from "../connectors/1inch.sol";
 
 contract MockConnectOne is ConnectOne {
   address public oneProtoAddr;