diff --git a/stegano/lsbset/generators.py b/stegano/lsbset/generators.py
index fe9a288..9c173a3 100644
--- a/stegano/lsbset/generators.py
+++ b/stegano/lsbset/generators.py
@@ -29,7 +29,7 @@ import itertools
 import cv2
 import numpy as np
 import math
-from typing import Dict, Iterator, List
+from typing import Dict, Iterator, List, Any, Union
 
 
 def identity() -> Iterator[int]:
@@ -230,14 +230,15 @@ def LFSR(m: int) -> Iterator[int]:
 
 
 def shi_tomashi(
-    image: str, corners: int = 100, quality: float = 0.01, min_distance: float = 10.0
+    image_path: str, corners: int = 100, quality: float = 0.01, min_distance: float = 10.0
 ) -> Iterator[int]:
     """Shi-Tomachi corner generator of the given points
     https://docs.opencv.org/4.x/d4/d8c/tutorial_py_shi_tomasi.html
     """
-    image = cv2.imread(image)
+    image = cv2.imread(image_path)
     gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    corners = cv2.goodFeaturesToTrack(gray, corners, quality, min_distance)
+    corners: Union[int, List[Any]] = cv2.goodFeaturesToTrack(
+        gray, corners, quality, min_distance)
     corners = np.int0(corners)
     i = 0
     while True: