Updated unit tests to work again.