Fix a couple of mistakes