Fix braces and a comment