Correction bug

Déplacement du code dans la bonne fonction.