make tests work in pycharm