PR comments

pooltogether · Oct 14, 2021 · b4d0b5b · b4d0b5b
1 parent 335dc47
commit b4d0b5b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/contracts/prize-pool/YieldSourcePrizePool.sol b/contracts/prize-pool/YieldSourcePrizePool.sol
@@ -66,9 +66,10 @@ contract YieldSourcePrizePool is PrizePool {
     /// @param _externalToken The address of the token to check
     /// @return True if the token may be awarded, false otherwise
     function _canAwardExternal(address _externalToken) internal view override returns (bool) {
+        IYieldSource _yieldSource = yieldSource;
         return (
-            _externalToken != address(yieldSource) &&
-            _externalToken != yieldSource.depositToken()
+            _externalToken != address(_yieldSource) &&
+            _externalToken != _yieldSource.depositToken()
         );
     }
 

diff --git a/test/prize-pool/YieldSourcePrizePool.test.ts b/test/prize-pool/YieldSourcePrizePool.test.ts
@@ -141,6 +141,10 @@ describe('YieldSourcePrizePool', function () {
         it('should not allow the prize pool to award its token, as its likely the receipt', async () => {
             expect(await prizePool.canAwardExternal(yieldSource.address)).to.equal(false);
         });
+
+        it('should not allow the prize pool to award the deposit token', async () =>  {
+            expect(await prizePool.canAwardExternal(depositToken.address)).to.equal(false);
+        })
     });
 
     describe('sweep()', () => {