Correction bug Code