diff --git a/uixt/android_test.go b/uixt/android_test.go
index c7cb883d..b8a085be 100644
--- a/uixt/android_test.go
+++ b/uixt/android_test.go
@@ -21,6 +21,11 @@ func setupADBDriverExt(t *testing.T) *XTDriver {
 		Serial:   "", // Let it auto-detect the device serial
 		AIOptions: []option.AIServiceOption{
 			option.WithCVService(option.CVServiceTypeVEDEM),
+			option.WithLLMConfig(
+				option.NewLLMServiceConfig(option.DOUBAO_1_5_UI_TARS_250328).
+					WithPlannerModel(option.WINGS_SERVICE).
+					WithAsserterModel(option.WINGS_SERVICE),
+			),
 		},
 	}
 
diff --git a/uixt/driver_ext_ai_test.go b/uixt/driver_ext_ai_test.go
index 90b0dcdb..6ca5d2e3 100644
--- a/uixt/driver_ext_ai_test.go
+++ b/uixt/driver_ext_ai_test.go
@@ -292,14 +292,31 @@ func TestDriverExt_AIAction(t *testing.T) {
 func TestDriverExt_AIAction_CompareWithAIAction(t *testing.T) {
 	driver := setupDriverExt(t)
 
-	prompt := "[目标导向]打开抖音，点击搜索按钮，搜索张杰，进入内容页停止"
+	prompt := "点击搜索按钮"
 
 	// Test both methods with the same prompt
-	aiResult, aiErr := driver.StartToGoal(context.Background(), prompt)
+	aiResult, aiErr := driver.AIAction(context.Background(), prompt)
 
 	// Both should execute without critical errors (may have different implementations)
 	t.Logf("AIAction error: %v", aiErr)
-	t.Logf("AIAction result: %v", aiResult)
+
+	// If both succeed, compare results
+	if aiResult != nil {
+		assert.Equal(t, "action", aiResult.Type, "AIAction result type should be 'action'")
+
+		// Both should have timing information
+		assert.Greater(t, aiResult.ModelCallElapsed, int64(0), "AIAction should have model call elapsed time")
+
+		// Both should have screenshot information
+		assert.NotEmpty(t, aiResult.ImagePath, "AIAction should have image path")
+
+		// Compare model names
+		if aiResult.PlanningResult != nil {
+			t.Logf("AIAction model: %s", aiResult.PlanningResult.ModelName)
+
+			assert.Equal(t, "wings-api", aiResult.PlanningResult.ModelName, "AIAction should use wings-api")
+		}
+	}
 }
 
 // TestDriverExt_AIAction_ErrorHandling tests AIAction error handling