diff --git a/cmd/agent/main.go b/cmd/agent/main.go
index 5ebf09f..2f5e084 100644
--- a/cmd/agent/main.go
+++ b/cmd/agent/main.go
@@ -35,7 +35,7 @@ func main() {
 	apiAddr := fmt.Sprintf("%s:%d", cfg.Api.Address, cfg.Api.Port)
 	promAddr := fmt.Sprintf("%s:%d", cfg.Prometheus.Address, cfg.Prometheus.Port)
 
-	go agentapi.New(q).Start(apiAddr)
+	go agentapi.New(q, db).Start(apiAddr)
 	go promserver.Start(promAddr, registry)
 
 	select {}
diff --git a/internal/api/agent/server.go b/internal/api/agent/server.go
index 75833c9..7e4eb21 100644
--- a/internal/api/agent/server.go
+++ b/internal/api/agent/server.go
@@ -5,14 +5,16 @@ import (
 	"net/http"
 
 	"git.g3e.fr/syonad/two/pkg/worker"
+	"github.com/dgraph-io/badger/v4"
 )
 
 type Server struct {
 	queue *worker.Queue
+	db    *badger.DB
 }
 
-func New(queue *worker.Queue) *Server {
-	return &Server{queue: queue}
+func New(queue *worker.Queue, db *badger.DB) *Server {
+	return &Server{queue: queue, db: db}
 }
 
 func (s *Server) Start(address string) {