fixing a couple small issues