@gkhachik do you think we should move this method to ```lib/cli/base.py``` and avoid importing the base class here?