fixed a few things I missed