Fixed test class compilation failure