Correction bug fix