diff --git a/client/models.go b/client/models.go
index e8a1b19..8756a60 100644
--- a/client/models.go
+++ b/client/models.go
@@ -21,19 +21,7 @@ type AssignTaskResponse struct {
 	Message        string  `json:"message"`
 	RateLimitDelay float64 `json:"rate_limit_delay,omitempty"`
 	Content        struct {
-		Task *struct {
-			Id                int64              `json:"id"`
-			Priority          int16              `json:"priority"`
-			Project           *storage.Project   `json:"project"`
-			Assignee          int64              `json:"assignee"`
-			Retries           int16              `json:"retries"`
-			MaxRetries        int64              `json:"max_retries"`
-			Status            storage.TaskStatus `json:"status"`
-			Recipe            string             `json:"recipe"`
-			MaxAssignTime     int64              `json:"max_assign_time"`
-			AssignTime        int64              `json:"assign_time"`
-			VerificationCount int16              `json:"verification_count"`
-		} `json:"task"`
+		storage.Task `json:"task"`
 	} `json:"content"`
 }