diff --git a/internal/task/handler.go b/internal/task/handler.go
index eae255c..1d01394 100644
--- a/internal/task/handler.go
+++ b/internal/task/handler.go
@@ -30,6 +30,7 @@ func New(deps Deps) *Handler {
 	}
 
 	taskTransportV1.RegisterTaskProcessorServer(srv, handler)
+	handler.srv = srv
 
 	return handler
 }