Unit tests work again