Fix test failure caused by missed rename