Petit refacto du code