diff --git a/internal/services/task.go b/internal/services/task.go
index 34cc538..dc6b2ae 100644
--- a/internal/services/task.go
+++ b/internal/services/task.go
@@ -1,25 +1,25 @@
 package services
 
 import (
-	"context"
-	"git.sch9.ru/new_gate/ms-tester/internal/models"
+    "context"
+    "git.sch9.ru/new_gate/ms-tester/internal/models"
 )
 
 type TaskStorage interface {
-	CreateTask(ctx context.Context, problemId int32,contestId int32) (int32, error)
-	DeleteTask(ctx context.Context, id int32) error
+    CreateTask(ctx context.Context, models.Task) (int32, error)
+    DeleteTask(ctx context.Context, id int32) error
 }
 
 type TaskService struct {
-	taskStorage TaskStorage
+    taskStorage TaskStorage
 }
 
 func NewTaskService(
-	taskStorage TaskStorage,
+    taskStorage TaskStorage,
 ) *TaskService {
-	return &TaskService{
-		taskStorage: taskStorage,
-	}
+    return &TaskService{
+        taskStorage: taskStorage,
+    }
 }
 
 func (service *TaskService) CreateTask(ctx context.Context, task models.Task) (int32, error) {