From 3b0999dcfafe6ce9073f87e16e7cb0ffc0739738 Mon Sep 17 00:00:00 2001
From: Schlag <89420541+Schlagonia@users.noreply.github.com>
Date: Mon, 5 Jun 2023 14:12:34 -0600
Subject: [PATCH] chore: factory init (#169)

---
 contracts/VaultFactory.vy | 5 ++---
 tests/conftest.py         | 5 ++++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/contracts/VaultFactory.vy b/contracts/VaultFactory.vy
index 83eaec78..2ea35ea8 100644
--- a/contracts/VaultFactory.vy
+++ b/contracts/VaultFactory.vy
@@ -29,7 +29,6 @@
     The protocol fees will be sent to the designated fee_recipient and
     then (X - protocol_fees) will be sent to the vault/strategy specific
     fee recipient.
-
 """
 
 from vyper.interfaces import ERC20
@@ -96,10 +95,10 @@ custom_protocol_fee: public(HashMap[address, uint16])
 use_custom_protocol_fee: public(HashMap[address, bool])
 
 @external
-def __init__(name: String[64], vault_blueprint: address):
+def __init__(name: String[64], vault_blueprint: address, governance: address):
     self.name = name
     VAULT_BLUEPRINT = vault_blueprint
-    self.governance = msg.sender
+    self.governance = governance
 
 @external
 def deploy_new_vault(
diff --git a/tests/conftest.py b/tests/conftest.py
index bb3e0f5c..d1537f8c 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -159,7 +159,10 @@ def vault_blueprint(project, gov):
 @pytest.fixture(scope="session")
 def vault_factory(project, gov, vault_blueprint):
     return gov.deploy(
-        project.VaultFactory, "Vault V3 Factory 3.0.1-beta", vault_blueprint
+        project.VaultFactory,
+        "Vault V3 Factory 3.0.1-beta",
+        vault_blueprint,
+        gov.address,
     )