Fix test for Python 3.x. D'oh