Add missing function argument