Minor fix

- Minor fix