A small bug fix