diff --git a/src/Blue.sol b/src/Blue.sol index 1fd5ae259..e5fc6f33e 100644 --- a/src/Blue.sol +++ b/src/Blue.sol @@ -73,7 +73,7 @@ contract Blue is IFlashLender { // Only owner functions. - function transferOwnership(address newOwner) external onlyOwner { + function setOwner(address newOwner) external onlyOwner { owner = newOwner; } diff --git a/test/forge/Blue.t.sol b/test/forge/Blue.t.sol index df02e303e..826484629 100644 --- a/test/forge/Blue.t.sol +++ b/test/forge/Blue.t.sol @@ -135,7 +135,7 @@ contract BlueTest is Blue blue2 = new Blue(oldOwner); vm.prank(oldOwner); - blue2.transferOwnership(newOwner); + blue2.setOwner(newOwner); assertEq(blue2.owner(), newOwner, "owner"); } @@ -146,7 +146,7 @@ contract BlueTest is vm.prank(attacker); vm.expectRevert(bytes(Errors.NOT_OWNER)); - blue2.transferOwnership(newOwner); + blue2.setOwner(newOwner); } function testEnableIrmWhenNotOwner(address attacker, IIrm newIrm) public {