Fix improper function reference