Fix a few small mistakes in the tests