Add missing braces