Fix Python 3+