Fixed broken case statement