fixes from pycharm