Make tests runnable in pycharm (almost good)