1 files changed, 2 insertions, 2 deletions

diff --git a/db/mongodb/client.go b/db/mongodb/client.go
index 705b151..c423d99 100644
--- a/db/mongodb/client.go
+++ b/db/mongodb/client.go
@@ -24,10 +24,10 @@ type Mongo struct {
         db     *mongo.Database
 }
 
-func Connect(ctx context.Context, uri, materialSet string) (*Mongo, error) {
+func Connect(ctx context.Context, uri, materialSet string, vc vault.VaultClient) (*Mongo, error) {
         db := &Mongo{}
 
-        cred, err := vault.GetVaultDatabase(materialSet)
+        cred, err := vc.DbCredential(ctx, materialSet)
         if err != nil {
                 return nil, err
         }