Fix bug, normalize PSF at the right time (before transforming it)