Replace label and break with defer and return