Fix test compliation