diff --git a/apps/desktop/src/__tests__/integration/template-weight-config.e2e.test.tsx b/apps/desktop/src/__tests__/integration/template-weight-config.e2e.test.tsx
index 1172720..b5d3365 100644
--- a/apps/desktop/src/__tests__/integration/template-weight-config.e2e.test.tsx
+++ b/apps/desktop/src/__tests__/integration/template-weight-config.e2e.test.tsx
@@ -119,7 +119,7 @@ describe('Template Weight Configuration E2E Tests', () => {
     jest.clearAllMocks();
     
     // Setup default mock implementations
-    mockAiClassificationService.getAllAiClassifications.mockResolvedValue(mockAiClassifications);
+    mockAiClassificationService.getAllClassifications.mockResolvedValue(mockAiClassifications);
     mockTemplateSegmentWeightService.getSegmentWeightsWithDefaults.mockResolvedValue({
       classification_1: 80,
       classification_2: 60,
diff --git a/apps/desktop/src/components/template/BatchWeightConfigModal.tsx b/apps/desktop/src/components/template/BatchWeightConfigModal.tsx
index 2d154af..5f4aff7 100644
--- a/apps/desktop/src/components/template/BatchWeightConfigModal.tsx
+++ b/apps/desktop/src/components/template/BatchWeightConfigModal.tsx
@@ -6,7 +6,7 @@ import {
   CheckIcon,
   ExclamationTriangleIcon,
 } from '@heroicons/react/24/outline';
-import { AiClassification } from '../../types/ai-classification';
+import { AiClassification } from '../../types/aiClassification';
 import { TemplateSegmentWeightHelper } from '../../types/template';
 import { TemplateSegmentWeightService } from '../../services/templateSegmentWeightService';
 import { AiClassificationService } from '../../services/aiClassificationService';
@@ -60,7 +60,7 @@ export const BatchWeightConfigModal: React.FC<BatchWeightConfigModalProps> = ({
 
   const loadAiClassifications = async () => {
     try {
-      const classifications = await AiClassificationService.getAllAiClassifications();
+      const classifications = await AiClassificationService.getAllClassifications();
       const activeClassifications = classifications.filter(c => c.is_active);
       setAiClassifications(activeClassifications);
       
diff --git a/apps/desktop/src/components/template/TemplateSegmentWeightEditor.tsx b/apps/desktop/src/components/template/TemplateSegmentWeightEditor.tsx
index cc958b4..e853a78 100644
--- a/apps/desktop/src/components/template/TemplateSegmentWeightEditor.tsx
+++ b/apps/desktop/src/components/template/TemplateSegmentWeightEditor.tsx
@@ -6,8 +6,8 @@ import {
   XMarkIcon,
   InformationCircleIcon 
 } from '@heroicons/react/24/outline';
-import { AiClassification } from '../../types/ai-classification';
-import { TemplateSegmentWeight, TemplateSegmentWeightHelper } from '../../types/template';
+import { AiClassification } from '../../types/aiClassification';
+import { TemplateSegmentWeightHelper } from '../../types/template';
 import { TemplateSegmentWeightService } from '../../services/templateSegmentWeightService';
 import { AiClassificationService } from '../../services/aiClassificationService';
 
@@ -57,7 +57,7 @@ export const TemplateSegmentWeightEditor: React.FC<TemplateSegmentWeightEditorPr
       setError(null);
 
       const [classifications, weights, hasCustom] = await Promise.all([
-        AiClassificationService.getAllAiClassifications(),
+        AiClassificationService.getAllClassifications(),
         TemplateSegmentWeightService.getSegmentWeightsWithDefaults(templateId, trackSegmentId),
         TemplateSegmentWeightService.hasCustomSegmentWeights(templateId, trackSegmentId),
       ]);
diff --git a/apps/desktop/src/components/template/__tests__/TemplateSegmentWeightEditor.test.tsx b/apps/desktop/src/components/template/__tests__/TemplateSegmentWeightEditor.test.tsx
index 01955bc..e85c2e4 100644
--- a/apps/desktop/src/components/template/__tests__/TemplateSegmentWeightEditor.test.tsx
+++ b/apps/desktop/src/components/template/__tests__/TemplateSegmentWeightEditor.test.tsx
@@ -63,7 +63,7 @@ describe('TemplateSegmentWeightEditor', () => {
     jest.clearAllMocks();
     
     // Setup default mock implementations
-    mockAiClassificationService.getAllAiClassifications.mockResolvedValue(mockAiClassifications);
+    mockAiClassificationService.getAllClassifications.mockResolvedValue(mockAiClassifications);
     mockTemplateSegmentWeightService.getSegmentWeightsWithDefaults.mockResolvedValue(mockCurrentWeights);
     mockTemplateSegmentWeightService.hasCustomSegmentWeights.mockResolvedValue(false);
   });
@@ -264,7 +264,7 @@ describe('TemplateSegmentWeightEditor', () => {
 
   it('handles loading state', () => {
     // Mock loading state
-    mockAiClassificationService.getAllAiClassifications.mockImplementation(
+    mockAiClassificationService.getAllClassifications.mockImplementation(
       () => new Promise(() => {}) // Never resolves
     );
 
@@ -274,7 +274,7 @@ describe('TemplateSegmentWeightEditor', () => {
   });
 
   it('handles error state', async () => {
-    mockAiClassificationService.getAllAiClassifications.mockRejectedValue(
+    mockAiClassificationService.getAllClassifications.mockRejectedValue(
       new Error('Failed to load classifications')
     );