Fix a couple of small mistakes