[FIX] small bug