Photon 4-momentum correction

following the vertex assignment