making test work again