Fix a problem with the unit tests.