small bug fixed, 2