Fix test I accidentally broke