Fix braces bug