nit: use braces around the `if` block