Python 2.x test fix