Fix missing function reference