Fix for missing function argument