nit: add curly braces please.