Nit fixes
Nit fixes