nit: Add curly brackets