fixed a couple small issues