small bug fix!