Small code modif