Just a small bug fix.