From 47931b3dd13c7dc63d4e545ca8f2c8145e499867 Mon Sep 17 00:00:00 2001
From: Simon Gockner <simon.gockner@gmail.com>
Date: Fri, 22 Nov 2019 14:46:13 +0100
Subject: [PATCH] #29: fix return value of `WithFactoryMethod()`

---
 .../Interfaces/Registrations/ISingleTypeRegistration.cs         | 2 +-
 LightweightIocContainer/Registrations/SingleTypeRegistration.cs | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/LightweightIocContainer/Interfaces/Registrations/ISingleTypeRegistration.cs b/LightweightIocContainer/Interfaces/Registrations/ISingleTypeRegistration.cs
index 8c68244..ad9f188 100644
--- a/LightweightIocContainer/Interfaces/Registrations/ISingleTypeRegistration.cs
+++ b/LightweightIocContainer/Interfaces/Registrations/ISingleTypeRegistration.cs
@@ -22,6 +22,6 @@ namespace LightweightIocContainer.Interfaces.Registrations
         /// </summary>
         /// <param name="factoryMethod">The <see cref="Func{T,TResult}"/></param>
         /// <returns>The current instance of this <see cref="IRegistrationBase{TInterface}"/></returns>
-        IRegistrationBase<T> WithFactoryMethod(Func<IIocContainer, T> factoryMethod);
+        ISingleTypeRegistration<T> WithFactoryMethod(Func<IIocContainer, T> factoryMethod);
     }
 }
\ No newline at end of file
diff --git a/LightweightIocContainer/Registrations/SingleTypeRegistration.cs b/LightweightIocContainer/Registrations/SingleTypeRegistration.cs
index 5a7a793..7a961c3 100644
--- a/LightweightIocContainer/Registrations/SingleTypeRegistration.cs
+++ b/LightweightIocContainer/Registrations/SingleTypeRegistration.cs
@@ -35,7 +35,7 @@ namespace LightweightIocContainer.Registrations
         /// </summary>
         /// <param name="factoryMethod">The <see cref="Func{T,TResult}"/></param>
         /// <returns>The current instance of this <see cref="IRegistrationBase{TInterface}"/></returns>
-        public IRegistrationBase<T> WithFactoryMethod(Func<IIocContainer, T> factoryMethod)
+        public ISingleTypeRegistration<T> WithFactoryMethod(Func<IIocContainer, T> factoryMethod)
         {
             FactoryMethod = factoryMethod;
             return this;