make unit tests work again