Missing braces fixed