fixed a couple things I missed