Fix test I just broke