small bug fixed..