Fix arg check