making tests work again