Fix CLI arguments handling