Minor fix

- Fixing a reference error