diff --git a/hyakvnc/__main__.py b/hyakvnc/__main__.py
index 7dda050..dcee1b4 100644
--- a/hyakvnc/__main__.py
+++ b/hyakvnc/__main__.py
@@ -184,9 +184,7 @@ def get_session():
                 logger.debug(f"Could not get session info for job {job_id}: {e}")
             return None
 
-        sesh = repeat_until(
-            lambda x: get_session(), lambda x: x is not None, timeout=app_config.sbatch_post_timeout * 2
-        )
+        sesh = repeat_until(lambda: get_session(), lambda x: x is not None, timeout=app_config.sbatch_post_timeout * 2)
         if not sesh:
             logger.warning(f"No running VNC sessions found for job {job_id}. Canceling and exiting.")
             kill_self()