cleanup

Layr-Labs · Nov 29, 2023 · 38dafc6 · 38dafc6
1 parent b56976c
commit 38dafc6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/contracts/pods/EigenPod.sol b/src/contracts/pods/EigenPod.sol
@@ -426,8 +426,12 @@ contract EigenPod is IEigenPod, Initializable, ReentrancyGuardUpgradeable, Eigen
     }
 
 
-    function turnOnPartialWithdrawalProofSwitch() external onlyEigenPodOwner {
-        partialWithdrawalProofSwitch = true;
+    function flipPartialWithdrawalProofSwitch() external onlyEigenPodOwner {
+        if(partialWithdrawalProofSwitch) {
+            partialWithdrawalProofSwitch = false;
+        } else {
+            partialWithdrawalProofSwitch = true;
+        }
     }
 
     /*******************************************************************************