Add missed braces