BUG: fixe small bug