Minor fix

minor fix