Fix prompting for arguments on CLI