diff --git a/internal/data/db.go b/internal/data/db.go
index a18626a..1e5c45d 100644
--- a/internal/data/db.go
+++ b/internal/data/db.go
@@ -15,6 +15,6 @@ func NewDb(db *GormDb) *Db {
 	}
 }
 
-func (d *Db) DB(_ context.Context) *gorm.DB {
-	return d.db.Client
+func (d *Db) DB(ctx context.Context) *gorm.DB {
+	return d.db.Client.WithContext(ctx)
 }