diff --git a/frontend/src/api/services/index.ts b/frontend/src/api/services/index.ts
index bb3f7416..cbaeae47 100644
--- a/frontend/src/api/services/index.ts
+++ b/frontend/src/api/services/index.ts
@@ -29,6 +29,7 @@ export type {
   TrainingJobProgress,
   ModelMetrics,
   TrainingConfiguration,
+  TrainingJobStatus
 } from './trainingService';
 
 export type {
diff --git a/frontend/src/pages/onboarding.tsx b/frontend/src/pages/onboarding.tsx
index 8f8b50ed..86592c89 100644
--- a/frontend/src/pages/onboarding.tsx
+++ b/frontend/src/pages/onboarding.tsx
@@ -18,7 +18,8 @@ import {
   TenantUser,
   TenantCreate,
   TenantInfo ,
-  DataValidation
+  DataValidation,
+  TrainingJobStatus
 } from '@/api/services';
 
 
@@ -136,7 +137,7 @@ const OnboardingPage = () => {
           console.log('Starting training with config:', trainingConfig);
           
           // Start training via API
-          const trainingJob = await api.training.startTraining(trainingConfig);
+          const trainingJob: TrainingJobStatus = await api.training.startTraining(trainingConfig);
           
           // Update form data with training job ID
           setFormData(prev => ({ 
diff --git a/services/training/app/services/training_service.py b/services/training/app/services/training_service.py
index 8a863182..7dec211c 100644
--- a/services/training/app/services/training_service.py
+++ b/services/training/app/services/training_service.py
@@ -410,8 +410,9 @@ class TrainingService:
         try:
             # Call data service to get sales data
             async with httpx.AsyncClient() as client:
-                params = {
-                    "tenant_id": tenant_id
+                params = {}
+                headers = {
+                    "X-Tenant-ID": tenant_id
                 }
                 
                 if hasattr(request, 'start_date') and request.start_date:
@@ -426,6 +427,7 @@ class TrainingService:
                 response = await client.get(
                     f"{settings.DATA_SERVICE_URL}/api/v1/sales/",
                     params=params,
+                    headers=headers,
                     timeout=30.0
                 )