Fix CLI prototype