fix arg count check