nit: use `} else {`