Another minor bug fix