From 591cf53b7463bd1c9e1145a176b539f58f4b24c2 Mon Sep 17 00:00:00 2001
From: stonehenge <stonehenge@looksrare.org>
Date: Thu, 4 Jul 2024 18:29:46 +0400
Subject: [PATCH] fix: Get modifier from TestHelpers

---
 test/foundry/BlastYield.t.sol | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/test/foundry/BlastYield.t.sol b/test/foundry/BlastYield.t.sol
index 43d3cab..4067bbf 100644
--- a/test/foundry/BlastYield.t.sol
+++ b/test/foundry/BlastYield.t.sol
@@ -6,6 +6,7 @@ import {IOwnableTwoSteps} from "../../contracts/interfaces/IOwnableTwoSteps.sol"
 import {OwnableTwoSteps} from "../../contracts/OwnableTwoSteps.sol";
 import {BlastYield} from "../../contracts/BlastYield.sol";
 import {Test} from "../../lib/forge-std/src/Test.sol";
+import {TestHelpers} from "./utils/TestHelpers.sol";
 import {YieldMode as IBlast__YieldMode, GasMode as IBlast__GasMode} from "../../contracts/interfaces/IBlast.sol";
 import {YieldMode as IERC20Rebasing__YieldMode} from "../../contracts/interfaces/IERC20Rebasing.sol";
 
@@ -29,7 +30,7 @@ contract BlastYieldOwnableTwoSteps is BlastYield, OwnableTwoSteps {
     }
 }
 
-contract BlastYieldOwnableTwoSteps_Test is Test {
+contract BlastYieldOwnableTwoSteps_Test is Test, TestHelpers {
     MockBlastWETH private weth;
     MockBlastERC20 private usdb;
     MockBlastYield private mockBlastYield;
@@ -87,10 +88,4 @@ contract BlastYieldOwnableTwoSteps_Test is Test {
         vm.expectRevert(IOwnableTwoSteps.NotOwner.selector);
         blastYieldOwnableTwoSteps.claim(TREASURY, TREASURY);
     }
-
-    modifier asPrankedUser(address user) {
-        vm.startPrank(user);
-        _;
-        vm.stopPrank();
-    }
 }