Fix conditional and include `end`