Only check for MTLModel protocol conformance

Mantle · Oct 28, 2015 · 857c826 · 857c826
1 parent 66af2bf
commit 857c826
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Mantle/MTLJSONAdapter.m b/Mantle/MTLJSONAdapter.m
@@ -508,7 +508,7 @@ @implementation MTLJSONAdapter (ValueTransformers)
 		reverseBlock:^ NSDictionary * (id model, BOOL *success, NSError **error) {
 			if (model == nil) return nil;
 
-			if (![model isKindOfClass:MTLModel.class] || ![model conformsToProtocol:@protocol(MTLJSONSerializing)]) {
+			if (![model conformsToProtocol:@protocol(MTLModel)] || ![model conformsToProtocol:@protocol(MTLJSONSerializing)]) {
 				if (error != NULL) {
 					NSDictionary *userInfo = @{
 						NSLocalizedDescriptionKey: NSLocalizedString(@"Could not convert model object to JSON dictionary", @""),

diff --git a/Mantle/NSDictionary+MTLMappingAdditions.m b/Mantle/NSDictionary+MTLMappingAdditions.m
@@ -13,7 +13,7 @@
 @implementation NSDictionary (MTLMappingAdditions)
 
 + (NSDictionary *)mtl_identityPropertyMapWithModel:(Class)modelClass {
-	NSCParameterAssert([modelClass isSubclassOfClass:MTLModel.class]);
+	NSCParameterAssert([modelClass conformsToProtocol:@protocol(MTLModel)]);
 
 	NSArray *propertyKeys = [modelClass propertyKeys].allObjects;