Fix missing braces