Fix forgotten braces