From f394f74e03be92bcc12b5addaaafcb78ad62df89 Mon Sep 17 00:00:00 2001
From: PJ <peterjan.brone@gmail.com>
Date: Wed, 16 Mar 2022 13:12:48 +0100
Subject: [PATCH] Implement MR remarks

---
 database/database.go | 2 +-
 main.go              | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/database/database.go b/database/database.go
index 75e7cf2..c9fdb5a 100644
--- a/database/database.go
+++ b/database/database.go
@@ -170,7 +170,7 @@ func NewTestDB(ctx context.Context, dbName string, logger *logrus.Logger) *DB {
 	}
 
 	// create a context with timeout and purge the database
-	ctx, cancel := context.WithTimeout(context.Background(), MongoDefaultTimeout)
+	ctx, cancel := context.WithTimeout(ctx, MongoDefaultTimeout)
 	defer cancel()
 	err = db.Purge(ctx)
 	if err != nil {
diff --git a/main.go b/main.go
index 0655864..20b09b9 100644
--- a/main.go
+++ b/main.go
@@ -13,7 +13,6 @@ import (
 	"github.com/SkynetLabs/blocker/api"
 	"github.com/SkynetLabs/blocker/blocker"
 	"github.com/SkynetLabs/blocker/database"
-	"github.com/SkynetLabs/blocker/skyd"
 	"github.com/SkynetLabs/blocker/syncer"
 	"github.com/joho/godotenv"
 	"github.com/sirupsen/logrus"