Fix Python 3 error