Fix bug in logic for determining output arg