[FIX] Fixed test due to name change