fix up a couple of tests