Fixing a reference error (copy/paste!)