Fix momentumTensor calculation

Norm of momentumTensor should be initialized with zero