some fixes
diff --git a/wally/ssh_utils.py b/wally/ssh_utils.py
index 0f46b83..e566cbe 100644
--- a/wally/ssh_utils.py
+++ b/wally/ssh_utils.py
@@ -455,7 +455,7 @@
 
         while self.check_running() and time.time() < soft_end_of_wait_time:
             # time.sleep(soft_end_of_wait_time - time.time())
-            time.sleep(2)
+            time.sleep(time_till_check)
 
         while end_of_wait_time > time.time():
             time.sleep(time_till_check)