computeMomentumEquation fixed (TB, AH, TS)