diff --git a/src/ImageHorizonLibrary/recognition/_recognize_images.py b/src/ImageHorizonLibrary/recognition/_recognize_images.py
index b2ed9d5..c012cc9 100644
--- a/src/ImageHorizonLibrary/recognition/_recognize_images.py
+++ b/src/ImageHorizonLibrary/recognition/_recognize_images.py
@@ -241,10 +241,10 @@ def wait_for(self, reference_image, timeout=10):
 
         Returns Python tuple ``(x, y)`` of the coordinates.
         '''
-        stop_time = time() + int(timeout)
+        stop_time = int(time()) + int(timeout)
         location = None
         with self._suppress_keyword_on_failure():
-            while time() < stop_time:
+            while int(time()) <= stop_time:
                 try:
                     location = self._locate(reference_image, log_it=False)
                     break