fixing the last of the tests before we break it again