nit: add braces