correct photon 4 momentum