Fix for test in Python3 take two