Fixed a couple mistakes