CLI fixes