Fix argparse in align.py