Correct missing function arg