Fix argument count check