Add missing curly brackets