a couple small tweaks