Refactor.

darrencauthon · Apr 21, 2016 · e7d3977 · e7d3977
1 parent a08f542
commit e7d3977
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/AutoMoq/Mocking.cs b/src/AutoMoq/Mocking.cs
@@ -71,9 +71,11 @@ public MockCreationResult CreateANewMockObjectAndRegisterIt(Type type)
             var result = CreateAMockObjectFor(type);
             var mock = (Mock) result.MockObject;
 
-            var method = ioc.GetType().GetMethods()[2];
-            var registerMethod = method.MakeGenericMethod(type);
-            registerMethod.Invoke(ioc, new[] { mock.Object });
+            ioc.GetType()
+                .GetMethods()
+                .First(x => x.Name == "RegisterInstance" && x.IsGenericMethod)
+                .MakeGenericMethod(type)
+                .Invoke(ioc, new[] { mock.Object });
 
             ioc.Resolve<AutoMoqer>().SetMock(type, mock);
             return result;