Python 3 test fix