Fix missing "end"