small fixes from pycharm