Correction with calculating mass