tests pass on Python 2
fix to test function.