Good catch, I forgot about the `argparse` import.