diff --git a/core/internal/testutil/account.go b/core/internal/testutil/account.go
index a13d370f3..4282068c4 100644
--- a/core/internal/testutil/account.go
+++ b/core/internal/testutil/account.go
@@ -327,9 +327,10 @@ func (account *ComprehensiveTestAccount) GetKeysForProvider(ctx *context.Context
 	case schemas.HuggingFace:
 		return []schemas.Key{
 			{
-				Value:  os.Getenv("HUGGING_FACE_API_KEY"),
-				Models: []string{},
-				Weight: 1.0,
+				Value:          os.Getenv("HUGGING_FACE_API_KEY"),
+				Models:         []string{},
+				Weight:         1.0,
+				UseForBatchAPI: bifrost.Ptr(true),
 			},
 		}, nil
 	case schemas.Nebius:
diff --git a/core/providers/utils/html_response_handler_test.go b/core/providers/utils/html_response_handler_test.go
index a9b31d8a9..ab702f2da 100644
--- a/core/providers/utils/html_response_handler_test.go
+++ b/core/providers/utils/html_response_handler_test.go
@@ -212,7 +212,7 @@ func TestHandleProviderAPIErrorWithHTML(t *testing.T) {
 				</html>
 			`),
 			description:       "Should detect and handle HTML only after JSON parse fails",
-			expectedInMessage: "Internal Server Error",
+			expectedInMessage: "HTML response received from provider",
 		},
 		{
 			name:        "HTML 403 error - lazy detection",
@@ -226,8 +226,8 @@ func TestHandleProviderAPIErrorWithHTML(t *testing.T) {
 				</body>
 				</html>
 			`),
-			description:       "Should detect and extract message from HTML on parse failure",
-			expectedInMessage: "Forbidden",
+			description:       "Should detect HTML on parse failure",
+			expectedInMessage: "HTML response received from provider",
 		},
 		{
 			name:              "Invalid JSON with HTML fallback",