Fix missing curly braces