diff --git a/mtda/main.py b/mtda/main.py
index 6b83259e..49d89166 100644
--- a/mtda/main.py
+++ b/mtda/main.py
@@ -167,7 +167,7 @@ def config_set_power_timeout(self, timeout, session=None):
         self._power_timeout = timeout
         if timeout == 0:
             self._power_expiry = None
-        self._session_check()
+        self._session_check(session)
 
         self.mtda.debug(3, f"main.config_set_power_timeout(): {result}")
         return result
@@ -187,7 +187,7 @@ def config_set_session_timeout(self, timeout, session=None):
                 left = self._sessions[s] - now
                 if left > timeout:
                     self._sessions[s] = now + timeout
-        self._session_check()
+        self._session_check(session)
 
         self.mtda.debug(3, f"main.config_set_session_timeout(): {result}")
         return result