diff --git a/src/skimage-stubs/filters.pyi b/src/skimage-stubs/filters.pyi
index 31a5e8bff056cfe3ef5e18d1188606ec5bb4ed3e..92909ae3c546c584baefa95469d346c1024050b3 100644
--- a/src/skimage-stubs/filters.pyi
+++ b/src/skimage-stubs/filters.pyi
@@ -3,7 +3,7 @@
 Typing stubs for skimage.filters.
 """
 from typing import Optional, Union, Any
-from scientific_plots.types_ import Matrix
+from scientific_plots.types_ import Matrix, Vector
 
 
 def butterworth(
diff --git a/src/sklearn-stubs/ensemble.pyi b/src/sklearn-stubs/ensemble.pyi
index 3df757c844b5f668be76a2e41228cf8128f64ba1..55a9b209cf2075ee7a0c5169cbc243627f67154c 100644
--- a/src/sklearn-stubs/ensemble.pyi
+++ b/src/sklearn-stubs/ensemble.pyi
@@ -4,10 +4,9 @@ Typing stubs for sklearn.ensemble.
 """
 from __future__ import annotations
 
-from typing import Optional, Union, List, Tuple, Dict, TypeVar
+from typing import Optional, Union, List, Dict, TypeVar
 from scientific_plots.types_ import Matrix, Vector as Vector_
 
-from vector_type import Vector  # Assuming Vector is defined in vector_type module
 
 Vector = TypeVar("Vector", bound=Union[Vector_, Matrix])
 
@@ -21,7 +20,7 @@ class IsolationForest:
                  bootstrap: bool = False, 
                  n_jobs: Optional[int] = None, 
                  behaviour: str = 'deprecated', 
-                 random_state: Optional[Union[int, np.random.RandomState]] = None, 
+                 random_state: Optional[int] = None, 
                  verbose: int = 0, 
                  warm_start: bool = False) -> None: ...
     
@@ -45,18 +44,18 @@ class IsolationForest:
                    deep: bool = True) -> Dict[str, 
                                             Union[int, float, str, bool, None]]: ...
     
-    def set_params(self, **params) -> IsolationForest: ...
+    def set_params(self, params: dict[str, float]) -> IsolationForest: ...
 
     # Additional methods for compatibility with BaseEstimator and OutlierMixin
-    def __getstate__(self) -> dict: ...
-    def __setstate__(self, state: dict) -> None: ...
+    def __getstate__(self) -> dict[str, float]: ...
+    def __setstate__(self, state: dict[str, float]) -> None: ...
 
     # Properties
     @property
-    def estimators_(self) -> List: ...
+    def estimators_(self) -> List[float]: ...
 
     @property
-    def estimators_samples_(self) -> List: ...
+    def estimators_samples_(self) -> List[float]: ...
 
     @property
     def max_samples_(self) -> int: ...