where M is the nuclear mass. P0, P1 and Pa are nuclear momenta on the lower, upper and active PESs based on energy conservation. The phase correction can be combined with different decoherence schemes, and the original Ehrenfest term in the EOMs for electronic coefficients (i.e., Eq. ( )) can be directly modified as