fix tests in an awkward way