Fix function names

Doh