Fix that test I broke