Fixed missing braces