Fix bug with braces