make the unit tests work again