small bug fix ..