Fix argument checking.