Minor fix

- Fixing a reference error (copy/paste!)