Fix broken argument parsing