Fix argument to use store_true