Fix argument count test