One small bug fix.