Small fix in distribution functions calculation

Stupid typo from somewhere: instead of u it operated with u**2.