fix: GetEntryAssembly

capdiem · Sep 8, 2021 · ebb9df8 · ebb9df8
1 parent eeab0f2
commit ebb9df8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Extensions/MASA.Utils.Extensions.DependencyInjection/ServiceCollectionExtensions.cs b/src/Extensions/MASA.Utils.Extensions.DependencyInjection/ServiceCollectionExtensions.cs
@@ -11,7 +11,7 @@ public static class ServiceCollectionExtensions
     public static IServiceCollection AddServices(this IServiceCollection services, string suffix, bool autoFire)
     {
         Assembly
-            .GetCallingAssembly()
+            .GetEntryAssembly()!
             .GetTypes()
             .Where(t => t.Name.EndsWith(suffix))
             .ToList()
@@ -30,7 +30,7 @@ public static IServiceCollection AddServices<TService>(this IServiceCollection s
         var serviceType = typeof(TService);
 
         Assembly
-            .GetCallingAssembly()
+            .GetEntryAssembly()!
             .GetTypes()
             .Where(t => t.BaseType == serviceType)
             .ToList()