Fix bugs in cli module