-[mom2d] Fixed small bugs