Fix function bug
Another function reference error