diff --git a/tests/integration_test/__tests__/ai.test.js b/tests/integration_test/__tests__/ai.test.js
index 1c86df63..fbb72bc4 100644
--- a/tests/integration_test/__tests__/ai.test.js
+++ b/tests/integration_test/__tests__/ai.test.js
@@ -90,9 +90,9 @@ test('test AI text gen', async() => {
 		],
 		dialogue_history: dialogueHistory,
 		ai_agent: {
-			type: 'ai_agent_ask',
-			basic_text_multi: {
-				model: 'openai__gpt_3_5_turbo'
+			type: 'ai_agent_text_gen',
+			basic_gen: {
+				model: 'openai__gpt_3_5_turbo_16k'
 			}
 		}
 	});