Fixed missing break in switch statement that was causing issues