Fix broken function reference