diff --git a/db/model/proxy_test.go b/db/model/proxy_test.go
index ad02eb6..e7b644c 100644
--- a/db/model/proxy_test.go
+++ b/db/model/proxy_test.go
@@ -52,9 +52,10 @@
 )
 
 func init() {
-	if _, err := log.SetLogger(log.CONSOLE, 0, log.Fields{"instanceId": "proxy_test"}); err != nil {
-		log.With(log.Fields{"error": err}).Fatal("cannot setup logging")
-	}
+
+	log.AddPackage(log.JSON, log.ErrorLevel, nil)
+	log.UpdateAllLoggers(log.Fields{"instanceId": "proxy_test"})
+
 	defer log.CleanUp()
 
 	pt.Backend = NewBackend(pt.DbType, pt.DbHost, pt.DbPort, pt.DbTimeout, pt.DbPrefix)
diff --git a/db/model/transaction_test.go b/db/model/transaction_test.go
index 064b9ef..130a992 100644
--- a/db/model/transaction_test.go
+++ b/db/model/transaction_test.go
@@ -50,9 +50,10 @@
 )
 
 func init() {
-	if _, err := log.SetLogger(log.CONSOLE, 0, log.Fields{"instanceId": "transaction_test"}); err != nil {
-		log.With(log.Fields{"error": err}).Fatal("cannot setup logging")
-	}
+
+	log.AddPackage(log.JSON, log.ErrorLevel, nil)
+	log.UpdateAllLoggers(log.Fields{"instanceId": "transaction_test"})
+
 	defer log.CleanUp()
 
 	tx.Backend = NewBackend(tx.DbType, tx.DbHost, tx.DbPort, tx.DbTimeout, tx.DbPrefix)