Partial fix for #328.

erikdoe · Oct 22, 2018 · 0b809a5 · 0b809a5
1 parent 44ee441
commit 0b809a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Source/OCMock/OCClassMockObject.m b/Source/OCMock/OCClassMockObject.m
@@ -145,6 +145,7 @@ - (void)prepareClassForClassMethodMocking
     }];
 }
 
+
 - (void)setupForwarderForClassMethodSelector:(SEL)selector
 {
     SEL aliasSelector = OCMAliasForOriginalSelector(selector);
@@ -157,8 +158,8 @@ - (void)setupForwarderForClassMethodSelector:(SEL)selector
 
     Class metaClass = object_getClass(mockedClass);
     IMP forwarderIMP = [originalMetaClass instanceMethodForwarderForSelector:selector];
-    class_replaceMethod(metaClass, selector, forwarderIMP, types);
     class_addMethod(metaClass, aliasSelector, originalIMP, types);
+    class_replaceMethod(metaClass, selector, forwarderIMP, types);
 }