diff --git a/frontend/composables/useTeacherList.ts b/frontend/composables/useTeacherList.ts
index bdc38522..5ae75910 100644
--- a/frontend/composables/useTeacherList.ts
+++ b/frontend/composables/useTeacherList.ts
@@ -1,31 +1,33 @@
 import { reactive } from '@nuxtjs/composition-api'
+import { Category } from '~/domain/models/tasks/category'
 
-export const useTeacherList = (service: any) => {
+export const useTeacherList = (repository: any) => {
   const state = reactive({
     teacherList: []
   })
 
   const getTeacherList = async (projectId: string, exampleId: number) => {
-    state.teacherList = await service.list(projectId, exampleId)
+    state.teacherList = await repository.list(projectId, exampleId)
   }
 
   const removeTeacher = async (projectId: string, exampleId: number, teacherId: number) => {
-    await service.delete(projectId, exampleId, teacherId)
+    await repository.delete(projectId, exampleId, teacherId)
     await getTeacherList(projectId, exampleId)
   }
 
   const annotateLabel = async (projectId: string, exampleId: number, labelId: number) => {
-    await service.create(projectId, exampleId, labelId)
+    const category = Category.create(labelId)
+    await repository.create(projectId, exampleId, category)
     await getTeacherList(projectId, exampleId)
   }
 
   const clearTeacherList = async (projectId: string, exampleId: number) => {
-    await service.clear(projectId, exampleId)
+    await repository.clear(projectId, exampleId)
     await getTeacherList(projectId, exampleId)
   }
 
   const autoLabel = async (projectId: string, exampleId: number) => {
-    await service.autoLabel(projectId, exampleId)
+    await repository.autoLabel(projectId, exampleId)
     await getTeacherList(projectId, exampleId)
   }
 
diff --git a/frontend/pages/projects/_id/text-classification/index.vue b/frontend/pages/projects/_id/text-classification/index.vue
index a382c3ad..cddbb212 100644
--- a/frontend/pages/projects/_id/text-classification/index.vue
+++ b/frontend/pages/projects/_id/text-classification/index.vue
@@ -42,19 +42,19 @@
 </template>
 
 <script>
-import { toRefs, useContext, useFetch, ref, watch } from '@nuxtjs/composition-api'
-import LabelGroup from '@/components/tasks/textClassification/LabelGroup'
-import LabelSelect from '@/components/tasks/textClassification/LabelSelect'
+import { ref, toRefs, useContext, useFetch, watch } from '@nuxtjs/composition-api'
 import LayoutText from '@/components/tasks/layout/LayoutText'
 import ListMetadata from '@/components/tasks/metadata/ListMetadata'
+import AnnotationProgress from '@/components/tasks/sidebar/AnnotationProgress.vue'
+import LabelGroup from '@/components/tasks/textClassification/LabelGroup'
+import LabelSelect from '@/components/tasks/textClassification/LabelSelect'
+import ButtonLabelSwitch from '@/components/tasks/toolbar/buttons/ButtonLabelSwitch'
 import ToolbarLaptop from '@/components/tasks/toolbar/ToolbarLaptop'
 import ToolbarMobile from '@/components/tasks/toolbar/ToolbarMobile'
-import ButtonLabelSwitch from '@/components/tasks/toolbar/buttons/ButtonLabelSwitch'
 import { useExampleItem } from '@/composables/useExampleItem'
 import { useLabelList } from '@/composables/useLabelList'
 import { useProjectItem } from '@/composables/useProjectItem'
 import { useTeacherList } from '@/composables/useTeacherList'
-import AnnotationProgress from '@/components/tasks/sidebar/AnnotationProgress.vue'
 
 export default {
   components: {
@@ -86,7 +86,7 @@ export default {
       clearTeacherList,
       getTeacherList,
       removeTeacher
-    } = useTeacherList(app.$services.textClassification)
+    } = useTeacherList(app.$repositories.category)
     const enableAutoLabeling = ref(false)
     const { state: labelState, getLabelList, shortKeys } = useLabelList(app.$services.categoryType)
     const labelComponent = ref('label-group')