Fix function referense (ugh missed it)