diff --git a/ConfectioneryContracts/DI/UnityDependencyContainer.cs b/ConfectioneryContracts/DI/UnityDependencyContainer.cs
index dd5e085..da703da 100644
--- a/ConfectioneryContracts/DI/UnityDependencyContainer.cs
+++ b/ConfectioneryContracts/DI/UnityDependencyContainer.cs
@@ -1,5 +1,7 @@
 using Microsoft.Extensions.Logging;
+using System.ComponentModel;
 using Unity;
+using Unity.Lifetime;
 using Unity.Microsoft.Logging;
 
 namespace ConfectioneryContracts.DI
@@ -11,13 +13,12 @@ namespace ConfectioneryContracts.DI
         public UnityDependencyContainer()
         {
             _container = new UnityContainer();
-            _container.AddExtension(new LoggingExtension());
         }
 
         public void AddLogging(Action<ILoggingBuilder> configure)
         {
-            var factory = _container.Configure<LoggingExtension>().LoggerFactory;
-            _container.Configure<LoggingExtension>();
+            var factory = LoggerFactory.Create(configure);
+            _container.AddExtension(new LoggingExtension(factory));
         }
 
         public void RegisterType<T>(bool isSingle) where T : class