Fix issue with missing break statement