Added missing braces