BUG: Correcting psi function