Add 'arg' check