diff --git a/src/google/adk/agents/base_agent.py b/src/google/adk/agents/base_agent.py
index bdc10ac3a..a4946a0eb 100644
--- a/src/google/adk/agents/base_agent.py
+++ b/src/google/adk/agents/base_agent.py
@@ -147,8 +147,7 @@ async def run_async(
       async for event in self._run_async_impl(ctx):
         yield event
 
-      if ctx.end_invocation:
-        return
+      # Don't check for end_invocation. Let the afer agent callbacks get called.
 
       if event := await self.__handle_after_agent_callback(ctx):
         yield event