Fixed test after changing enum