A minor bug fix