Add missing function call