Small modif in ui