Minor tweak to test for Python 3