Another small bug fix.