diff --git a/cvx/simulator/builder.py b/cvx/simulator/builder.py
index d2d9868b..ec28627b 100644
--- a/cvx/simulator/builder.py
+++ b/cvx/simulator/builder.py
@@ -113,7 +113,7 @@ def current_prices(self) -> np.array:
         """
         return self._state.prices[self._state.assets].values
 
-    def __iter__(self) -> Generator[tuple[pd.DatetimeIndex, State], None, None]:
+    def __iter__(self) -> Generator[tuple[pd.DatetimeIndex, State]]:
         """
         The __iter__ method allows the object to be iterated over in a for loop,
         yielding time and the current state of the portfolio.