Fix cli import