Adds missing braces