fix arg logic