diff --git a/src/app/api/tasks/index.ts b/src/app/api/tasks/index.ts
index 855a900..675ff1b 100644
--- a/src/app/api/tasks/index.ts
+++ b/src/app/api/tasks/index.ts
@@ -3,6 +3,7 @@ import { BASE_API_URL } from "@/config/api";
 import { cache } from "react";
 import "server-only";
 import { NewTaskTemplateFormInputs } from "./actions";
+import { Grade } from "../grades";
 
 export interface TaskGroup {
   id: number;
@@ -16,11 +17,23 @@ export interface Task {
   taskGroup: TaskGroup;
 }
 
+export interface TaskTemplateGradeAllocation {
+  grade: Grade;
+  percentage: number;
+}
+
+export interface TaskTemplateGroupAllocation {
+  taskGroup: TaskGroup;
+  percentage: number;
+}
+
 export interface TaskTemplate {
   id: number;
   code: string;
   name: string;
   tasks: Task[];
+  gradeAllocations: TaskTemplateGradeAllocation[];
+  groupAllocations: TaskTemplateGroupAllocation[];
 }
 
 export const preloadTaskTemplates = () => {