Clean up the base model arg