Fix a couple mistakes