Fix a couple errors