Fix error in argument parsing