Fixed a couple things from last commit.