Fix break stmt handling