fix a couple of tests