Minor bug with argparse arguments