little fix
refacto