diff --git a/openhands/runtime/impl/action_execution/action_execution_client.py b/openhands/runtime/impl/action_execution/action_execution_client.py
index 3ae72b7790fa..d9aa31d8c846 100644
--- a/openhands/runtime/impl/action_execution/action_execution_client.py
+++ b/openhands/runtime/impl/action_execution/action_execution_client.py
@@ -45,7 +45,7 @@
 from openhands.utils.tenacity_stop import stop_if_should_exit
 
 
-def _is_retryable_check_alive_error(exception):
+def _is_retryable_error(exception):
     return isinstance(
         exception, (httpx.RemoteProtocolError, httpcore.RemoteProtocolError)
     )
@@ -93,6 +93,11 @@ def __init__(
     def _get_action_execution_server_host(self) -> str:
         pass
 
+    @retry(
+        retry=retry_if_exception(_is_retryable_error),
+        stop=stop_after_attempt(5) | stop_if_should_exit(),
+        wait=wait_exponential(multiplier=1, min=4, max=15),
+    )
     def _send_action_server_request(
         self,
         method: str,
@@ -114,11 +119,6 @@ def _send_action_server_request(
         """
         return send_request(self.session, method, url, **kwargs)
 
-    @retry(
-        retry=retry_if_exception(_is_retryable_check_alive_error),
-        stop=stop_after_attempt(5) | stop_if_should_exit(),
-        wait=wait_exponential(multiplier=1, min=4, max=15),
-    )
     def check_if_alive(self) -> None:
         response = self._send_action_server_request(
             'GET',