From 47b096e7eff011f137e650073bb25108bd0db8d5 Mon Sep 17 00:00:00 2001
From: Samyak Jain <34437877+KaymasJain@users.noreply.github.com>
Date: Sat, 20 Jun 2020 18:01:45 +1000
Subject: [PATCH] edited small thing on COMP contract.

---
 contracts/connectors/compound.sol | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/contracts/connectors/compound.sol b/contracts/connectors/compound.sol
index f42b88e..1686b4a 100644
--- a/contracts/connectors/compound.sol
+++ b/contracts/connectors/compound.sol
@@ -132,10 +132,10 @@ contract CompoundHelpers is Helpers {
     }
 
     /**
-     * @dev Return COMP Token.
+     * @dev Return COMP Token Address.
      */
-    function getCompToken() internal pure returns (TokenInterface) {
-        return TokenInterface(0xc00e94Cb662C3520282E6f5717214004A7f26888);
+    function getCompTokenAddress() internal pure returns (address) {
+        return 0xc00e94Cb662C3520282E6f5717214004A7f26888;
     }
 
     /**
@@ -301,9 +301,10 @@ contract ExtraResolver is BasicResolver {
      * @param setId Set ctoken amount at this ID in `InstaMemory` Contract.
     */
     function ClaimComp(uint setId) external payable {
-        uint intialBal = getCompToken().balanceOf(address(this));
+        TokenInterface compToken = TokenInterface(getCompTokenAddress());
+        uint intialBal = compToken.balanceOf(address(this));
         ComptrollerInterface(getComptrollerAddress()).claimComp(address(this));
-        uint finalBal = getCompToken().balanceOf(address(this));
+        uint finalBal = compToken.balanceOf(address(this));
         uint amt = sub(finalBal, intialBal);
 
         setUint(setId, amt);