Little fix in CLI