diff --git a/src/common/Elsa.Api.Common/Extensions/ModuleExtensions.cs b/src/common/Elsa.Api.Common/Extensions/ModuleExtensions.cs index 0cb3cbfa1..3789578aa 100644 --- a/src/common/Elsa.Api.Common/Extensions/ModuleExtensions.cs +++ b/src/common/Elsa.Api.Common/Extensions/ModuleExtensions.cs @@ -15,21 +15,22 @@ public static class ModuleExtensions /// /// Registers the specified assembly for FastEndpoint assembly discovery. /// - public static void AddFastEndpointsAssembly(this IModule module, Assembly assembly) + public static IModule AddFastEndpointsAssembly(this IModule module, Assembly assembly) { var assemblies = module.Properties.GetOrAdd(FastEndpointsAssembliesKey, () => new HashSet()); assemblies.Add(assembly); + return module; } /// /// Registers the assembly for FastEndpoint assembly discovery using the specified marker type. /// - public static void AddFastEndpointsAssembly(this IModule module) => module.AddFastEndpointsAssembly(typeof(T)); + public static IModule AddFastEndpointsAssembly(this IModule module) => module.AddFastEndpointsAssembly(typeof(T)); /// /// Registers the assembly for FastEndpoint assembly discovery using the specified marker type. /// - public static void AddFastEndpointsAssembly(this IModule module, Type markerType) => module.AddFastEndpointsAssembly(markerType.Assembly); + public static IModule AddFastEndpointsAssembly(this IModule module, Type markerType) => module.AddFastEndpointsAssembly(markerType.Assembly); /// /// Returns all collected assemblies for discovery of endpoints.