small bug correction
again..