use ISamWitchVRF

PaintSwap · Feb 24, 2024 · 29ec761 · 29ec761
1 parent 06ae4fd
commit 29ec761
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/contracts/test/TestVRFConsumer.sol b/contracts/test/TestVRFConsumer.sol
@@ -2,14 +2,14 @@
 pragma solidity ^0.8.24;
 
 import {ISamWitchVRFConsumer} from "../interfaces/ISamWitchVRFConsumer.sol";
-import {SamWitchVRF} from "../SamWitchVRF.sol";
+import {ISamWitchVRF} from "../interfaces/ISamWitchVRF.sol";
 import {Ownable} from "@openzeppelin/contracts/access/Ownable.sol";
 
 contract TestVRFConsumer is ISamWitchVRFConsumer, Ownable {
   error ShouldRevertTrue();
   error OnlySamWitchVRF();
 
-  SamWitchVRF public samWitchVRF;
+  ISamWitchVRF public samWitchVRF;
   mapping(bytes32 requestId => uint256[] randomWords) public allRandomWords;
   bool private shouldRevert;
 
@@ -20,7 +20,7 @@ contract TestVRFConsumer is ISamWitchVRFConsumer, Ownable {
     _;
   }
 
-  constructor(SamWitchVRF _samWitchVRF) Ownable(msg.sender) {
+  constructor(ISamWitchVRF _samWitchVRF) Ownable(msg.sender) {
     samWitchVRF = _samWitchVRF;
   }