Minor fix

TBR