diff --git a/Probability/Projected.cs b/Probability/Projected.cs index 29aae74..559665d 100644 --- a/Probability/Projected.cs +++ b/Probability/Projected.cs @@ -14,8 +14,10 @@ public static IDiscreteDistribution Distribution( Func projection) { var result = new Projected(underlying, projection); - if (result.Support().Count() == 1) - return Singleton.Distribution(result.Support().First()); + var support = result.Support().ToList(); + + if (support.Count() == 1) + return Singleton.Distribution(support.Single()); return result; } private Projected(