Fixed a couple things I broke.