diff --git a/src/AttributeSourceGenerator/AttributeIncrementalGeneratorBase.cs b/src/AttributeSourceGenerator/AttributeIncrementalGeneratorBase.cs
index eeaaf84..3c240c2 100644
--- a/src/AttributeSourceGenerator/AttributeIncrementalGeneratorBase.cs
+++ b/src/AttributeSourceGenerator/AttributeIncrementalGeneratorBase.cs
@@ -1,4 +1,5 @@
-using System.Text;
+using System.Collections.Immutable;
+using System.Text;
 using AttributeSourceGenerator.Common;
 using AttributeSourceGenerator.Models;
 using Microsoft.CodeAnalysis;
@@ -36,8 +37,8 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
     {
         context.RegisterPostInitializationOutput(AddMarkerAttributeSource);
 
-        var syntaxProvider = context.SyntaxProvider.ForAttributeWithMetadataName(_configuration.MarkerAttributeName, Filter, Transform);
-        context.RegisterSourceOutput(syntaxProvider, GenerateSourceForSymbol);
+        var syntaxProvider = context.SyntaxProvider.ForAttributeWithMetadataName(_configuration.MarkerAttributeName, Filter, Transform).Collect();
+        context.RegisterSourceOutput(syntaxProvider, GenerateSourceForSymbols);
     }
 
     /// <summary>Adds the marker attribute source to the output.</summary>
@@ -129,12 +130,12 @@ private static Symbol Transform(GeneratorAttributeSyntaxContext context, Cancell
         return symbol;
     }
 
-    /// <summary>Generates source code for a given symbol.</summary>
+    /// <summary>Generates source for the given symbols.</summary>
     /// <param name="context">The source production context.</param>
-    /// <param name="symbol">The symbol to generate source for.</param>
-    private void GenerateSourceForSymbol(SourceProductionContext context, Symbol symbol)
+    /// <param name="symbols">The symbols to generate source for.</param>
+    private void GenerateSourceForSymbols(SourceProductionContext context, ImmutableArray<Symbol> symbols)
     {
-        var sources = _configuration.SourceGenerator(symbol);
+        var sources = _configuration.SourceGenerator(symbols);
         foreach (var source in sources)
             context.AddSource($"{source.Name}.g.cs", source.Text);
     }
diff --git a/src/AttributeSourceGenerator/AttributeIncrementalGeneratorConfiguration.cs b/src/AttributeSourceGenerator/AttributeIncrementalGeneratorConfiguration.cs
index 06425cf..dd1b37a 100644
--- a/src/AttributeSourceGenerator/AttributeIncrementalGeneratorConfiguration.cs
+++ b/src/AttributeSourceGenerator/AttributeIncrementalGeneratorConfiguration.cs
@@ -1,4 +1,6 @@
-namespace AttributeSourceGenerator;
+using System.Collections.Immutable;
+
+namespace AttributeSourceGenerator;
 
 /// <summary>Defines the configuration for an incremental attribute generator.</summary>
 public sealed class AttributeIncrementalGeneratorConfiguration
@@ -13,7 +15,7 @@ public sealed class AttributeIncrementalGeneratorConfiguration
     public FilterType SymbolFilter { get; init; } = FilterType.All;
 
     /// <summary>The function that generates the source code for the attribute.</summary>
-    public required Func<Symbol, IEnumerable<Source>> SourceGenerator { get; init; }
+    public required Func<ImmutableArray<Symbol>, IEnumerable<Source>> SourceGenerator { get; init; }
 
     /// <summary>Initializes a new instance of the <see cref="AttributeIncrementalGeneratorConfiguration" /> class</summary>
     public AttributeIncrementalGeneratorConfiguration()
diff --git a/src/AttributeSourceGenerator/AttributeSourceGenerator.csproj b/src/AttributeSourceGenerator/AttributeSourceGenerator.csproj
index bf3dbb9..73fc851 100644
--- a/src/AttributeSourceGenerator/AttributeSourceGenerator.csproj
+++ b/src/AttributeSourceGenerator/AttributeSourceGenerator.csproj
@@ -6,7 +6,7 @@
         <Nullable>enable</Nullable>
         <RootNamespace>AttributeSourceGenerator</RootNamespace>
         <LangVersion>latest</LangVersion>
-        <Version>8.0.3</Version>
+        <Version>8.0.4</Version>
         <Title>AttributeSourceGenerator</Title>
         <Authors>Jean-Sebastien Carle</Authors>
         <Description>A simple attribute-based Roslyn incremental source generator base class for .NET.</Description>
@@ -18,8 +18,8 @@
         <RepositoryUrl>https://github.com/jscarle/AttributeSourceGenerator</RepositoryUrl>
         <RepositoryType>git</RepositoryType>
         <PackageTags>attribute source-generator attribute-based source-generators</PackageTags>
-        <AssemblyVersion>8.0.3.0</AssemblyVersion>
-        <FileVersion>8.0.3.0</FileVersion>
+        <AssemblyVersion>8.0.4.0</AssemblyVersion>
+        <FileVersion>8.0.4.0</FileVersion>
         <NeutralLanguage>en-US</NeutralLanguage>
         <IncludeSymbols>true</IncludeSymbols>
         <SymbolPackageFormat>snupkg</SymbolPackageFormat>