fix for mass loss
Merge pull request #11 from marce155/master

fix for mass loss