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.