small bug fix..