nit: move this comment after the next `if`