Fixed a couple minor bugs.