Add missed brace