Fix color bug in PSF plot