Fix the problem with argparse