small fix
cleanup