Fix missing curly brackets