One small bug fix