Fix a couple of the tests