From 034584f6dee9144ffe4291f25de501283ed2bfc0 Mon Sep 17 00:00:00 2001
From: gpsanant <gpsanant@gmail.com>
Date: Sat, 12 Oct 2024 15:36:21 -0700
Subject: [PATCH] fix: add checks

---
 src/contracts/core/AVSDirectory.sol | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/contracts/core/AVSDirectory.sol b/src/contracts/core/AVSDirectory.sol
index 1c43c643a..05ea062a8 100644
--- a/src/contracts/core/AVSDirectory.sol
+++ b/src/contracts/core/AVSDirectory.sol
@@ -229,6 +229,7 @@ contract AVSDirectory is
         IStrategy[] calldata strategies
     ) external override {
         OperatorSet memory operatorSet = OperatorSet(msg.sender, operatorSetId);
+        require(isOperatorSet[msg.sender][operatorSetId], InvalidOperatorSet());
         bytes32 encodedOperatorSet = _encodeOperatorSet(operatorSet);
         for (uint256 i = 0; i < strategies.length; i++) {
             require(_operatorSetStrategies[encodedOperatorSet].add(address(strategies[i])), StrategyAlreadyInOperatorSet());
@@ -242,6 +243,7 @@ contract AVSDirectory is
         IStrategy[] calldata strategies
     ) external override {
         OperatorSet memory operatorSet = OperatorSet(msg.sender, operatorSetId);
+        require(isOperatorSet[msg.sender][operatorSetId], InvalidOperatorSet());
         bytes32 encodedOperatorSet = _encodeOperatorSet(operatorSet);
         for (uint256 i = 0; i < strategies.length; i++) {
             require(_operatorSetStrategies[encodedOperatorSet].remove(address(strategies[i])), StrategyNotInOperatorSet());