undo changes from d9d5a22

parallaxe · Jan 7, 2014 · 4ad7f97 · 4ad7f97
1 parent 761c3c3
commit 4ad7f97
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Source/OCMock/OCClassMockObject.m b/Source/OCMock/OCClassMockObject.m
@@ -25,15 +25,15 @@ + (void)rememberMock:(OCClassMockObject *)mock forClass:(Class)aClass
 {
     @synchronized(mockTable)
     {
-        [mockTable setObject:[NSValue valueWithNonretainedObject:mock] forKey:NSStringFromClass(aClass)];
+        [mockTable setObject:[NSValue valueWithNonretainedObject:mock] forKey:[NSValue valueWithNonretainedObject:aClass]];
     }
 }
 
 + (void)forgetMockForClass:(Class)aClass
 {
     @synchronized(mockTable)
     {
-        [mockTable removeObjectForKey:NSStringFromClass(aClass)];
+        [mockTable removeObjectForKey:[NSValue valueWithNonretainedObject:aClass]];
     }
 }
 
@@ -44,7 +44,7 @@ + (OCClassMockObject *)existingMockForClass:(Class)aClass
         OCClassMockObject *mock = nil;
         while((mock == nil) && (aClass != nil))
         {
-            mock = [[mockTable objectForKey:NSStringFromClass(aClass)] nonretainedObjectValue];
+            mock = [[mockTable objectForKey:[NSValue valueWithNonretainedObject:aClass]] nonretainedObjectValue];
             aClass = class_getSuperclass(aClass);
         }
         if(mock == nil)