fixed the unit tests to work again