diff --git a/src/services/apis/custom-api.mjs b/src/services/apis/custom-api.mjs
index b351ec2..e8fbaae 100644
--- a/src/services/apis/custom-api.mjs
+++ b/src/services/apis/custom-api.mjs
@@ -67,6 +67,7 @@ export async function generateAnswersWithCustomApi(port, question, session, apiK
         pushRecord(session, question, answer)
         console.debug('conversation history', { content: session.conversationRecords })
         port.postMessage({ answer: null, done: true, session: session })
+        return
       }
 
       if (data.response) answer = data.response
diff --git a/src/services/apis/openai-api.mjs b/src/services/apis/openai-api.mjs
index 4822c61..6eaee81 100644
--- a/src/services/apis/openai-api.mjs
+++ b/src/services/apis/openai-api.mjs
@@ -75,6 +75,7 @@ export async function generateAnswersWithGptCompletionApi(
         pushRecord(session, question, answer)
         console.debug('conversation history', { content: session.conversationRecords })
         port.postMessage({ answer: null, done: true, session: session })
+        return
       }
 
       answer += data.choices[0].text
@@ -170,6 +171,7 @@ export async function generateAnswersWithChatgptApiCompat(
         pushRecord(session, question, answer)
         console.debug('conversation history', { content: session.conversationRecords })
         port.postMessage({ answer: null, done: true, session: session })
+        return
       }
 
       const delta = data.choices[0]?.delta?.content