fixed a couple of small issues