Fixing Pmf.Max