Assorted code fixes

2 files changed, 3 insertions, 3 deletions

diff --git a/lockchain-core/src/init.rs b/lockchain-core/src/init.rs
index bf40518..0e666bb 100644
--- a/lockchain-core/src/init.rs
+++ b/lockchain-core/src/init.rs
@@ -75,6 +75,6 @@ impl Generator {
         V: Vault<B>,
         B: Body,
     {
-        V::new(self)
+        V::new(self).map(|b| *b)
     }
 }
diff --git a/lockchain-core/src/lib.rs b/lockchain-core/src/lib.rs
index 52139bb..bdcebdc 100644
--- a/lockchain-core/src/lib.rs
+++ b/lockchain-core/src/lib.rs
@@ -81,12 +81,12 @@ mod init;
 pub use self::crypto::PackedData;
 pub use self::meta::{MetaDomain, VaultMetadata};
 pub use self::record::{EncryptedBody, Header, Payload, Record};
-pub use self::init::Generator;
+pub use self::init::{VaultType, Generator};
 
 /// Export commonly used types via the prelude
 pub mod prelude {
     pub use super::crypto::PackedData;
     pub use super::meta::{MetaDomain, VaultMetadata};
     pub use super::record::{EncryptedBody, Header, Payload, Record};
-    pub use super::init::Generator;
+    pub use super::init::{VaultType, Generator};
 }