feat(protocol): Fix failing unit test (#14358)

Co-authored-by: David <[email protected]>
taikoxyz · Aug 3, 2023 · 42bbc5f · 42bbc5f
1 parent 1f2d4bc
commit 42bbc5f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/packages/protocol/test/ERC20Vault.t.sol b/packages/protocol/test/ERC20Vault.t.sol
@@ -387,11 +387,13 @@ contract TestERC20Vault is Test {
         // Upgrade the implementation of that contract
         // so that it supports now the 'helloWorld' call
         UpdatedBridgedERC20 newBridgedContract = new UpdatedBridgedERC20();
+        vm.stopPrank();
         vm.prank(Amelia, Amelia);
         TransparentUpgradeableProxy(payable(bridgedAddressAfter)).upgradeTo(
             address(newBridgedContract)
         );
 
+        vm.prank(Alice, Alice);
         try UpdatedBridgedERC20(bridgedAddressAfter).helloWorld() {
             //It should support now this function call
             assertEq(true, true);