Test for momentum: fixing comilation error