Skip to content

Commit

Permalink
update naming convention
Browse files Browse the repository at this point in the history
  • Loading branch information
gretzke committed Sep 1, 2023
1 parent b9c3cfe commit 59876e2
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/PolygonMigration.sol
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ contract PolygonMigration is Ownable2Step, IPolygonMigration {
uint256 public releaseTimestamp;
bool public unmigrationLocked;

modifier ifUnmigrationUnlocked() {
modifier onlyUnmigrationUnlocked() {
if (unmigrationLocked) revert UnmigrationLocked();
_;
}
Expand Down Expand Up @@ -54,7 +54,7 @@ contract PolygonMigration is Ownable2Step, IPolygonMigration {

/// @notice This function allows for unmigrating from POL tokens to MATIC tokens
/// @param amount Amount of POL to migrate
function unmigrate(uint256 amount) external ifUnmigrationUnlocked {
function unmigrate(uint256 amount) external onlyUnmigrationUnlocked {
emit Unmigrated(msg.sender, amount);

polygon.safeTransferFrom(msg.sender, address(this), amount);
Expand All @@ -67,7 +67,7 @@ contract PolygonMigration is Ownable2Step, IPolygonMigration {
function unmigrateTo(
uint256 amount,
address account
) external ifUnmigrationUnlocked {
) external onlyUnmigrationUnlocked {
emit Unmigrated(msg.sender, amount);

polygon.safeTransferFrom(msg.sender, address(this), amount);
Expand All @@ -82,7 +82,7 @@ contract PolygonMigration is Ownable2Step, IPolygonMigration {
uint8 v,
bytes32 r,
bytes32 s
) external ifUnmigrationUnlocked {
) external onlyUnmigrationUnlocked {
emit Unmigrated(msg.sender, amount);

IERC20Permit(address(polygon)).safePermit(
Expand Down

0 comments on commit 59876e2

Please sign in to comment.