Add argument guard