Fix Python 3 syntax