Fix nit in code