Fixed some things I missed