Fix Only to work on PyCharm