On the form of the momentum equation for shallow water models based on the generalized wave continuity equation