fixing up psi transformation