diff --git a/Classes/CoreDataManager.h b/Classes/CoreDataManager.h
index 6be4502..e1aaf9e 100644
--- a/Classes/CoreDataManager.h
+++ b/Classes/CoreDataManager.h
@@ -23,6 +23,7 @@
 
 - (BOOL)saveContext;
 - (void)useInMemoryStore;
+- (void)setDefaultModelFromClass:(Class)klass;
 
 #pragma mark - Helpers
 
diff --git a/Classes/CoreDataManager.m b/Classes/CoreDataManager.m
index cca8bbd..e521a69 100644
--- a/Classes/CoreDataManager.m
+++ b/Classes/CoreDataManager.m
@@ -97,6 +97,11 @@ - (BOOL)saveContext {
     return YES;
 }
 
+- (void)setDefaultModelFromClass:(Class)klass {
+    NSBundle *bundle = [NSBundle bundleForClass:klass];
+    _managedObjectModel = [NSManagedObjectModel mergedModelFromBundles:[NSArray arrayWithObject:bundle]];
+}
+
 
 #pragma mark - SQLite file directory