Fixed failing test in Python 3.x