Fix argument checking ...