[FIX] Fixed a small bug