Fix bug: Add missing breaks to switch