Fix 'missing braces' warning