Restore missing braces