Make test work again