From 46eb02c1204e09af84ebc63ed58ebf8c7cd603f5 Mon Sep 17 00:00:00 2001
From: Matthias Beyer <mail@beyermatthias.de>
Date: Fri, 20 Jan 2017 15:26:36 +0100
Subject: [PATCH] Impl Wrap for EntryHeader

---
 libimagruby/src/store.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/libimagruby/src/store.rs b/libimagruby/src/store.rs
index 827c3ce6..2a465084 100644
--- a/libimagruby/src/store.rs
+++ b/libimagruby/src/store.rs
@@ -211,13 +211,19 @@ pub mod store {
 
         wrappable_struct!(EntryHeader, EntryHeaderWrapper, ENTRY_HEADER_WRAPPER);
         class!(REntryHeader);
+
+        impl Wrap for EntryHeader {
+            fn wrap(self) -> AnyObject {
+                Class::from_existing("REntryHeader").wrap_data(self, &*ENTRY_HEADER_WRAPPER)
+            }
+        }
+
         methods!(
             REntryHeader,
             itself,
 
             fn r_entry_header_new() -> AnyObject {
-                Class::from_existing("REntryHeader")
-                    .wrap_data(EntryHeader::new(), &*ENTRY_HEADER_WRAPPER)
+                EntryHeader::new().wrap()
             }
 
             fn r_entry_header_insert(spec: RString, obj: AnyObject) -> Boolean {