Correct missing braces