Fix parse_args not expecting sys.argv as input