fix a couple small things