Repaired failing test on Python 3