Just minor cleanup

minor cleanup