diff --git a/.github/workflows/send-message-integration-tests.yaml b/.github/workflows/send-message-integration-tests.yaml
index 795614a9..e578bad4 100644
--- a/.github/workflows/send-message-integration-tests.yaml
+++ b/.github/workflows/send-message-integration-tests.yaml
@@ -148,6 +148,7 @@ jobs:
           DEEPSEEK_API_KEY: ${{ secrets.DEEPSEEK_API_KEY }}
           GOOGLE_CLOUD_PROJECT: ${{ secrets.GOOGLE_CLOUD_PROJECT }}
           GOOGLE_CLOUD_LOCATION: ${{ secrets.GOOGLE_CLOUD_LOCATION }}
+          LETTA_GEMINI_FORCE_MINIMUM_THINKING_BUDGET: true
         run: |
           poetry run pytest \
             -s -vv \
diff --git a/tests/integration_test_send_message.py b/tests/integration_test_send_message.py
index e0798841..a8250e85 100644
--- a/tests/integration_test_send_message.py
+++ b/tests/integration_test_send_message.py
@@ -817,6 +817,8 @@ def test_agent_loop_error(
             agent_id=agent_state_no_tools.id,
             messages=USER_MESSAGE_FORCE_REPLY,
         )
+
+    time.sleep(0.5)
     messages_from_db = client.agents.messages.list(agent_id=agent_state_no_tools.id, after=last_message[0].id)
     assert len(messages_from_db) == 0