fix up tests from 609a270f to also work in python 3