diff --git a/management/internals/network_map_db/db_store.go b/management/internals/network_map_db/db_store.go
index a5401635a..f1ca00826 100644
--- a/management/internals/network_map_db/db_store.go
+++ b/management/internals/network_map_db/db_store.go
@@ -46,7 +46,7 @@ type NetworkMapDBStoreImpl struct {
 	extraSettingsManager    settings.Manager
 }
 
-func NewNetworkMapDBStoreImpl(store NetworkMapDBStore, integratedPeerValidator integrated_validator.IntegratedValidator, extraSettingsManager integrations.Manager) *NetworkMapDBStoreImpl {
+func NewNetworkMapDBStoreImpl(store NetworkMapDBStore, integratedPeerValidator integrated_validator.IntegratedValidator, extraSettingsManager settings.Manager) *NetworkMapDBStoreImpl {
 	return &NetworkMapDBStoreImpl{
 		store:                   store,
 		integratedPeerValidator: integratedPeerValidator,
diff --git a/management/internals/network_map_db/pgsql/pg_store.go b/management/internals/network_map_db/pgsql/pg_store.go
index 9c7b98682..e18a0c830 100644
--- a/management/internals/network_map_db/pgsql/pg_store.go
+++ b/management/internals/network_map_db/pgsql/pg_store.go
@@ -23,7 +23,7 @@ type PgStore struct {
 }
 
 func NewPostgresqlStore(ctx context.Context, dsn string) (*PgStore, error) {
-	pool, err := connectToPgDb(context.Background(), dsn)
+	pool, err := connectToPgDb(ctx, dsn)
 	if err != nil {
 		return nil, err
 	}