Fixed two places where args was used instead of arg as a variable