Fix arg error