Momentum balance in the shallow water equations on bottom discontinuities