Fix argument check