fix issues

src/OpStateBridge.sol

@@ -14,6 +14,7 @@ import {ICrossDomainOwnable3} from "./interfaces/ICrossDomainOwnable3.sol";
 /// @author Worldcoin
 /// @notice Distributes new World ID Identity Manager roots to an OP Stack network
 /// @dev This contract lives on Ethereum mainnet and works for Optimism and any OP Stack based chain
+/// @dev Ownable2Step allows for transferOwnership to the zero address
 contract OpStateBridge is Ownable2Step {
     ///////////////////////////////////////////////////////////////////
     ///                           STORAGE                           ///

src/PolygonStateBridge.sol

@@ -13,6 +13,7 @@ import {FxBaseRootTunnel} from "fx-portal/contracts/tunnel/FxBaseRootTunnel.sol"
 /// @notice Distributes new World ID Identity Manager roots to World ID supported networks
 /// @dev This contract lives on Ethereum mainnet and is called by the World ID Identity Manager contract
 /// in the registerIdentities method
+/// @dev Ownable2Step allows for transferOwnership to the zero address
 contract PolygonStateBridge is FxBaseRootTunnel, Ownable2Step {
     ///////////////////////////////////////////////////////////////////
     ///                           STORAGE                           ///

src/PolygonWorldID.sol

@@ -13,6 +13,7 @@ import {BytesUtils} from "./utils/BytesUtils.sol";
 /// @notice A contract that manages the root history of the WorldID merkle root on Polygon PoS.
 /// @dev This contract is deployed on Polygon PoS and is called by the StateBridge contract for each
 ///      new root insertion.
+/// @dev Ownable2Step allows for transferOwnership to the zero address
 contract PolygonWorldID is WorldIDBridge, FxBaseChildTunnel, Ownable2Step {
     ///////////////////////////////////////////////////////////////////
     ///                           STORAGE                           ///

src/test/OpStateBridge.t.sol

@@ -139,8 +139,6 @@ contract OpStateBridgeTest is PRBTest, StdCheats {
     /// using Ownable2Step transferOwnership
     /// @param newOwner the new owner of the contract
     function test_owner_transferOwnership_succeeds(address newOwner) public {
-        vm.assume(newOwner != address(0));
-
         vm.expectEmit(true, true, true, true);
 
         // OpenZeppelin Ownable2Step transferOwnershipStarted event
@@ -164,8 +162,6 @@ contract OpStateBridgeTest is PRBTest, StdCheats {
     /// @param newOwner The new owner of the OPWorldID contract (foundry fuzz)
     /// @param isLocal Whether the ownership transfer is local (Optimism EOA/contract) or an Ethereum EOA or contract
     function test_owner_transferOwnershipOp_succeeds(address newOwner, bool isLocal) public {
-        vm.assume(newOwner != address(0));
-
         vm.expectEmit(true, true, true, true);
 
         // CrossDomainOwnable3.sol transferOwnership event
@@ -256,18 +252,6 @@ contract OpStateBridgeTest is PRBTest, StdCheats {
         opStateBridge.transferOwnershipOp(newOwner, isLocal);
     }
 
-    /// @notice tests that the StateBridge contract's ownership can't be changed by a non-owner
-    function test_newOwnerZero_transferOwnershipOp_reverts(
-        bool isLocal
-    ) public {
-        address newOwner = address(0);
-
-        vm.expectRevert("Ownable: new owner is the zero address");
-
-        vm.prank(owner);
-        opStateBridge.transferOwnershipOp(newOwner, isLocal);
-    }
-
     /// @notice tests whether the StateBridge contract can set root history expiry on Optimism and Polygon
     /// @param _rootHistoryExpiry The new root history expiry for OpWorldID and PolygonWorldID
     function test_notOwner_SetRootHistoryExpiry_reverts(

src/test/OpWorldID.t.sol

@@ -242,7 +242,7 @@ contract OpWorldIDTest is Messenger_Initializer {
             0,
             0,
             abi.encodeWithSelector(id.receiveRoot.selector, newRoot)
-        );        
+        );
 
         // It reverts with CannotOverwriteRoot however because of the bridge simulation
         // the L2 cross-domain call doesn't revert, however it does emit a FailedRelayedMessage

src/test/PolygonStateBridge.t.sol

@@ -106,8 +106,6 @@ contract PolygonStateBridgeTest is PRBTest, StdCheats {
     /// using Ownable2Step transferOwnership
     /// @param newOwner the new owner of the contract
     function test_owner_transferOwnership_succeeds(address newOwner) public {
-        vm.assume(newOwner != address(0));
-
         vm.expectEmit(true, true, true, true);
 
         // OpenZeppelin Ownable2Step transferOwnershipStarted event

src/test/PolygonWorldID.t.sol

@@ -45,8 +45,6 @@ contract PolygonWorldIDTest is PRBTest, StdCheats {
     /// using Ownable2Step transferOwnership
     /// @param newOwner the new owner of the contract
     function test_owner_transferOwnership_succeeds(address newOwner) public {
-        vm.assume(newOwner != address(0));
-
         vm.prank(owner);
         id.transferOwnership(newOwner);