From b290cb852d39754b8beb6662968f7a48aff48299 Mon Sep 17 00:00:00 2001
From: Kamran Sadin <MrSadin@Gmail.Com>
Date: Sat, 21 Jun 2025 11:08:45 +0330
Subject: [PATCH] Fix DI

---
 .../KSMessaging/Extensions/RegisterMediatorServices.cs        | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs b/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs
index 92c6fca..b2bba17 100644
--- a/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs
+++ b/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs
@@ -20,6 +20,8 @@ public static IServiceCollection AddKSMediator(this IServiceCollection services,
     {
         services.AddScoped<IMediator, Mediator>();
         services.AddScoped<ISender>(sp => sp.GetRequiredService<IMediator>());
+        
+        services.RegisterAllImplementationsOf<IRequestDecorator>(assemblies);
 
         services.Scan(scan => scan
             .FromAssemblies(assemblies)
@@ -72,8 +74,6 @@ public static IServiceCollection AddKSMediator(this IServiceCollection services,
 
         // If you have default behaviors (e.g., logging), register them here
         services.AddScoped(typeof(IStreamPipelineBehavior<,>), typeof(StreamLoggingBehavior<,>));
-        
-        services.RegisterAllImplementationsOf<IRequestDecorator>(assemblies);
 
         return services;
     }