Python 3 compatability change