diff --git a/src/ManiaMap.Tests/Generators/TestLayoutGraphSelector.cs b/src/ManiaMap.Tests/Generators/TestLayoutGraphSelector.cs
index 846713d7..7f854e61 100644
--- a/src/ManiaMap.Tests/Generators/TestLayoutGraphSelector.cs
+++ b/src/ManiaMap.Tests/Generators/TestLayoutGraphSelector.cs
@@ -41,7 +41,7 @@ public void TestFunctionSelector()
         {
             var seed = new RandomSeed(12345);
 
-            var graphs = new List<Func<LayoutGraph>>()
+            var graphs = new List<LayoutGraphSelector.LayoutGraphDelegate>()
             {
                 () => new LayoutGraph(1, "Graph1"),
                 () => new LayoutGraph(2, "Graph2"),
diff --git a/src/ManiaMap/Generators/LayoutGraphSelector.cs b/src/ManiaMap/Generators/LayoutGraphSelector.cs
index 0282c87c..fff3a081 100644
--- a/src/ManiaMap/Generators/LayoutGraphSelector.cs
+++ b/src/ManiaMap/Generators/LayoutGraphSelector.cs
@@ -12,6 +12,11 @@ namespace MPewsey.ManiaMap.Generators
     /// </summary>
     public class LayoutGraphSelector : IPipelineStep
     {
+        /// <summary>
+        /// A delegate returning a LayoutGraph.
+        /// </summary>
+        public delegate LayoutGraph LayoutGraphDelegate();
+        
         /// <summary>
         /// Draws a random layout graph and adds a copy to the results output dictionary.
         /// 
@@ -54,7 +59,7 @@ public LayoutGraph DrawSelection(IList<LayoutGraph> graphs, RandomSeed randomSee
         /// <param name="functions">A list of functions returning a layout graph.</param>
         /// <param name="randomSeed">The random seed.</param>
         /// <param name="logger">The logging action. Ignored if null.</param>
-        public LayoutGraph DrawSelection(IList<Func<LayoutGraph>> functions, RandomSeed randomSeed, Action<string> logger = null)
+        public LayoutGraph DrawSelection(IList<LayoutGraphDelegate> functions, RandomSeed randomSeed, Action<string> logger = null)
         {
             logger?.Invoke("[Layout Graph Selector] Applying layout graph delegate list selector...");
             var index = randomSeed.Next(0, functions.Count);
@@ -78,7 +83,7 @@ public LayoutGraph DrawSelection(object graphs, RandomSeed randomSeed, Action<st
             {
                 case IList<LayoutGraph> list:
                     return DrawSelection(list, randomSeed, logger);
-                case IList<Func<LayoutGraph>> functions:
+                case IList<LayoutGraphDelegate> functions:
                     return DrawSelection(functions, randomSeed, logger);
                 default:
                     throw new ArgumentException($"Unhandled type for `graphs`: {graphs.GetType()}.");